Update modeling_prot2text.py

modeling_prot2text.py

@@ -327,9 +327,9 @@ class Prot2TextModel(PreTrainedModel):
                                             output_scores=False, 
                                             return_dict_in_generate=True, 
                                             encoder_attention_mask=inputs['attention_mask'], 
-                                            length_penalty=2.0,
-                                            no_repeat_ngram_size=3,
-                                            early_stopping=True,
+                                            length_penalty=1.0,
+                                            no_repeat_ngram_size=None,
+                                            early_stopping=False,
                                             num_beams=1)
 
             generated = tokenizer.batch_decode(tok_ids.get('sequences'), skip_special_tokens=True)