adding library name

facebookresearch · Nov 6, 2023 · f0b0e80 · f0b0e80
1 parent 8dc2c95
commit f0b0e80
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/audiocraft/models/loaders.py b/audiocraft/models/loaders.py
@@ -27,6 +27,7 @@
 from omegaconf import OmegaConf, DictConfig
 import torch
 
+import audiocraft
 from . import builders
 from .encodec import CompressionModel
 
@@ -60,7 +61,9 @@ 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)
+        file = hf_hub_download(
+            repo_id=file_or_url_or_id, filename=filename, cache_dir=cache_dir,
+            library_name="audiocraft", library_version=audiocraft.__version__)
         return torch.load(file, map_location=device)