up

facebookresearch · Oct 30, 2023 · 4aec50a · 4aec50a
1 parent 029b6d6
commit 4aec50a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/audiocraft/models/loaders.py b/audiocraft/models/loaders.py
@@ -26,6 +26,7 @@
 
 from omegaconf import OmegaConf, DictConfig
 import torch
+import audiocraft
 
 from . import builders
 from .encodec import CompressionModel
@@ -60,7 +61,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", library_version="1.0.0")
+        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)