add library_version.

facebookresearch · Oct 19, 2023 · 029b6d6 · 029b6d6
1 parent 85473bb
commit 029b6d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)