From 4aec50a294d0078afcbbfc93f6017843b0815ba5 Mon Sep 17 00:00:00 2001 From: Vaibhav Srivastav Date: Mon, 30 Oct 2023 18:27:29 +0100 Subject: [PATCH] up --- audiocraft/models/loaders.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/audiocraft/models/loaders.py b/audiocraft/models/loaders.py index c11cc1e7..9411a913 100644 --- 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)