[bhavanatumma/ta2614748] - signature fix

Azure · Mar 7, 2024 · 2dab0dc · 2dab0dc
1 parent 8f324c5
commit 2dab0dc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...ts/training/model_management/src/azureml/model/mgmt/processors/transformers/convertors.py b/...ts/training/model_management/src/azureml/model/mgmt/processors/transformers/convertors.py
@@ -517,7 +517,8 @@ def save_as_mlflow(self):
         )
         self._hf_config_cls = self._hf_config_cls if self._hf_config_cls else AutoConfig
         self._hf_tokenizer_cls = self._hf_tokenizer_cls if self._hf_tokenizer_cls else AutoTokenizer
-        self._signatures = self._signatures or self.get_model_signature()
+        if self._model_flavor == "OSS":
+            self._signatures = self._signatures or self.get_model_signature()
 
         hf_conf[HF_CONF.HF_CONFIG_CLASS.value] = self._hf_config_cls.__name__
         hf_conf[HF_CONF.HF_PRETRAINED_CLASS.value] = self._hf_model_cls.__name__