hezarai
/

vit-roberta-fa-base

Model card Files Files and versions

arxyzan commited on Dec 20, 2023

Commit

b7e5be7

·

1 Parent(s): a05776e

Update model_config.yaml

Files changed (1) hide show

model_config.yaml +1 -1

model_config.yaml CHANGED Viewed

@@ -44,7 +44,7 @@ decoder:
 generation:
   bos_token_id: 0
   decoder_start_token_id: 0
-  return_dict_in_generate: true
   early_stopping: true
   eos_token_id: 2
   length_penalty: 2.0

 generation:
   bos_token_id: 0
   decoder_start_token_id: 0
+  return_dict_in_generate: false
   early_stopping: true
   eos_token_id: 2
   length_penalty: 2.0