fix bug

modeling_songformer.py

@@ -96,15 +96,15 @@ class SongFormerModel(PreTrainedModel):
     def __init__(self, config: SongFormerConfig):
         super().__init__(config)
         device = "cpu"
-
-        with open("muq_config2.json", "r") as f:
+        root_dir = os.path.dirname(os.path.abspath(__file__))
+        with open(os.path.join(root_dir, "muq_config2.json"), "r") as f:
             muq_config_file = OmegaConf.load(f)
         # self.muq = MuQ.from_pretrained("OpenMuQ/MuQ-large-msd-iter", device_map=None)
         self.muq = MuQ(muq_config_file)
 
         self.musicfm = MusicFM25Hz(
             is_flash=False,
-            stat_path="msd_stats.json",
+            stat_path=os.path.join(root_dir, "msd_stats.json"),
             # model_path=os.path.join(MUSICFM_HOME_PATH, "pretrained_msd.pt"),
         )
         self.songformer = Model(ModelConfig())