diff --git a/audiocraft/models/loaders.py b/audiocraft/models/loaders.py index 37e829d3..c11cc1e7 100644 --- a/audiocraft/models/loaders.py +++ b/audiocraft/models/loaders.py @@ -60,7 +60,7 @@ def _get_state_dict( else: assert filename is not None, "filename needs to be defined if using HF checkpoints" - file = hf_hub_download(repo_id=file_or_url_or_id, filename=filename, cache_dir=cache_dir, library_name="audiocraft") + file = hf_hub_download(repo_id=file_or_url_or_id, filename=filename, cache_dir=cache_dir, library_name="audiocraft", library_version="1.0.0") return torch.load(file, map_location=device)