ASLP-lab
/

SongFormer

music-structure-annotation

Model card Files Files and versions

ASLP-lab commited on Oct 10, 2025

Commit

b0d4960

·

1 Parent(s): b1682c1

fix bug

Files changed (1) hide show

modeling_songformer.py +1 -1

modeling_songformer.py CHANGED Viewed

@@ -96,7 +96,7 @@ class SongFormerModel(PreTrainedModel):
     def __init__(self, config: SongFormerConfig):
         super().__init__(config)
         device = "cpu"
-        root_dir = os.path.dirname(__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)

     def __init__(self, config: SongFormerConfig):
         super().__init__(config)
         device = "cpu"
+        root_dir = os.environ["SONGFORMER_LOCAL_DIR"]
         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)