utils.add_libraries_to_model creates a duplicated model

pablobd
Contributor II

Hello,

When I call this function,

mlflow.models.utils.add_libraries_to_model(MODEL_URI)

It register a new model into the Model Registry. Is it possible to do the same but without registering a new model?

Thanks,

pablobd
Contributor II

Thanks @Retired_mod - that's exactly how I am using it, following this: https://docs.databricks.com/en/machine-learning/model-serving/private-libraries-model-serving.html

What I don't like is that in step 2 the model is registered then in step 3, the same model (with the compiled package is registered again). So, I endup with two copies of the same model.

pablobd
Contributor II

I ended up publishing the library to AWS CodeArtifact repository. Now, how can I tell MLFlow to use AWS CodeArtifact private repository instead of PyPi?