24 lines
822 B
Python
24 lines
822 B
Python
def load_from_hub(repo_id: str, filename: str) -> str:
|
|
"""
|
|
Download a model from Hugging Face Hub.
|
|
:param repo_id: id of the model repository from the Hugging Face Hub
|
|
:param filename: name of the model zip file from the repository
|
|
"""
|
|
try:
|
|
from huggingface_hub import hf_hub_download
|
|
except ImportError:
|
|
raise ImportError(
|
|
"You need to install huggingface_hub to use `load_from_hub`. "
|
|
"See https://pypi.org/project/huggingface-hub/ for installation."
|
|
)
|
|
|
|
# Get the model from the Hub, download and cache the model on your local disk
|
|
downloaded_model_file = hf_hub_download(
|
|
repo_id=repo_id,
|
|
filename=filename,
|
|
library_name="huggingface-sb3",
|
|
library_version="2.1",
|
|
)
|
|
|
|
return downloaded_model_file
|