Support multiple models per repo as different models in the catalog? #323
Replies: 2 comments
-
At present, you would add a repo with >1 models to the catalog, then do: load_pretrained_model('path/to/repo', model='model_name') where 'model_name' is what comes under the So we could keep it this way, if we want to distinguish models based on setting up a class for each in the model.py of the model repo - this would save having to add to the catalog multiple times |
Beta Was this translation helpful? Give feedback.
-
Perhaps we could add an extra column to the returned dataframe from compatible_models, which is a list of the possible options for the |
Beta Was this translation helpful? Give feedback.
-
Currently we have one catalog entry per model repository, as oppose to one entry per model. Should we allow the catalog to have multiple entries that link to the same GitHub repo, but refer to different models within it?
Beta Was this translation helpful? Give feedback.
All reactions