Merge pull request #67 from boun-tabi-LMG/gokceuludogan-patch-fix-pre…

…dictor Set early_stopping False and fix TextPredictor's predict
boun-tabi-LMG · Feb 5, 2024 · cee3f7a · cee3f7a
2 parents e068908 + ab7a410
commit cee3f7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/turkish_lm_tuner/predictor.py b/turkish_lm_tuner/predictor.py
@@ -9,7 +9,7 @@ class TaskConfig:
     max_new_tokens: int = None
     length_penalty: float = None
     no_repeat_ngram_size: int = None
-    early_stopping: bool = True
+    early_stopping: bool = False
     decoder_start_token_id: int = None
     eos_token_id: int = None
     pad_token_id: int = None
@@ -142,6 +142,6 @@ def __init__(self, model_name, task, task_format='conditional_generation', max_i
         self.task_config = TaskConfig(**task_parameters[task])
 
     def predict(self, text, **kwargs):
-        generation_config = vars(self.task_config, **kwargs) if self.task_format == 'conditional_generation' else {}
+        generation_config = {**vars(self.task_config), **kwargs} if self.task_format == 'conditional_generation' else {}
         return super().predict(text, generation_config)