try registering just the model name

mosaicml · Oct 26, 2024 · 6c5fb05 · 6c5fb05
1 parent 4327257
commit 6c5fb05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -180,7 +180,7 @@ def save_model_patch(*args: Any, **kwargs: Any):
             input_example=input_example,
             metadata=log_model_metadata,
             task=task,
-            registered_model_name=mlflow_logger.model_registry_prefix + registered_model_name,
+            registered_model_name=registered_model_name, # not the full path? mlflow_logger.model_registry_prefix
             await_creation_for=await_creation_for
         )
     else: