Fixes issue with translate and temp models (#805)

Author: TaperChipmunk32

silnlp/nmt/experiment.py

@@ -92,7 +92,10 @@ def translate(self):
 
         for config in translate_configs.get("translate", []):
             translator = TranslationTask(
-                name=self.name, checkpoint=config.get("checkpoint", "last"), commit=self.commit
+                name=self.name,
+                checkpoint=config.get("checkpoint", "last"),
+                save_checkpoints=self.save_checkpoints,
+                commit=self.commit,
             )
 
             # Backwards compatibility

silnlp/nmt/hugging_face_config.py

@@ -1700,7 +1700,7 @@ def _create_inference_model(
             checkpoint_path, _ = self.get_checkpoint_path(ckpt)
             model_name = str(checkpoint_path)
         else:
-            LOGGER.warn("Model has no checkpoints. Using base model.")
+            LOGGER.warning("Model has no checkpoints. Using base model.")
             model_name = self._config.model
 
         dtype = torch.bfloat16 if self._is_t5 else torch.float16

silnlp/nmt/translate.py

@@ -41,6 +41,7 @@ def translate(
 class TranslationTask:
     name: str
     checkpoint: Union[str, int] = "last"
+    save_checkpoints: bool = False
     clearml_queue: Optional[str] = None
     commit: Optional[str] = None
 
@@ -266,6 +267,7 @@ def _init_translation_task(self, experiment_suffix: str) -> Tuple[Translator, Co
             project_suffix="_infer",
             experiment_suffix=experiment_suffix,
             commit=self.commit,
+            use_default_model_dir=self.save_checkpoints,
         )
         self.name = clearml.name