Update MuCodec/generate.py

MuCodec/generate.py

@@ -12,6 +12,7 @@ import numpy as np
 from tools.get_melvaehifigan48k import build_pretrained_models
 import tools.torch_tools as torch_tools
 from safetensors.torch import load_file
+from cached_path import cached_path
 
 class MuCodec:
     def __init__(self, \
@@ -26,7 +27,7 @@ class MuCodec:
 
         self.MAX_DURATION = 360
         if load_main_model:
-            audio_ldm_path = os.path.dirname(os.path.abspath(__file__)) + "/tools/audioldm_48k.pth"
+            audio_ldm_path = str(cached_path("hf://haoheliu/audioldm_48k/audioldm_48k.pth"))
             self.vae, self.stft = build_pretrained_models(audio_ldm_path)
             self.vae, self.stft = self.vae.eval().to(device), self.stft.eval().to(device)
             main_config = {