mb.get_models(...)
Retrieves multiple models from the registry.
Parameters
prefix
:Optional[str]
A search prefix to use when looking for models to fetch. Similar tomb.models(prefix=...)
.names
:Optional[List[str]]
A list of model names to fetch from the registry.files
:Optional[Dict[str, str]]
A dict of model names to fetch from the registry, and their corresponding local files paths for storage.branch
:Optional[s]
The branch to use when fetching models from the registry. By default, the current branch is used.
Returns
Dict[str, Any]
- A dict of model names to model instances, when using prefix=
or names=
. Or None
when using files=
.
Examples
Get several models by name
models = mb.get_models(names=["model1", "model2"])
Get several models by prefix
models = mb.get_models(prefix="project_foo/")
Download several models to files
mb.get_models(files={"model1": "path/to/model1.pkl", "model2": "path/to/model2.gguf"})