update

inspiremusic/cli/inference.py

@@ -266,7 +266,7 @@ def get_args():
     args = parser.parse_args()
 
     if not args.model_dir:
-        args.model_dir = os.path.join("../../pretrained_models", args.model_name)
+        args.model_dir = os.path.join("pretrained_models", args.model_name)
 
     print(args)
     return args

inspiremusic/flow/flow.py

@@ -39,7 +39,7 @@ class MaskedDiff(torch.nn.Module):
                                                           'n_blocks': 4, 'num_mid_blocks': 12, 'num_heads': 8, 'act_fn': 'gelu'}},
                  mel_feat_conf: Dict = {'n_fft': 1024, 'num_mels': 128, 'sampling_rate': 48000,
                                         'hop_size': 256, 'win_size': 1024, 'fmin': 0, 'fmax': 48000},
-                generator_model_dir: str = "../../pretrained_models/InspireMusic-Base/music_tokenizer",
+                generator_model_dir: str = "pretrained_models/InspireMusic-Base/music_tokenizer",
                 num_codebooks: int = 4
                 ):
         super().__init__()