jadechoghari
/

openmusic

Model card Files Files and versions Community

jadechoghari commited on 8 days ago

Commit

bf7634c

•

1 Parent(s): 76089fe

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -30,7 +30,7 @@ class MOSDiffusionPipeline(DiffusionPipeline):
         if self.reload_from_ckpt is not None:
             self.configs["reload_from_ckpt"] = self.reload_from_ckpt
-        self.dataset_key = self.build_dataset_json_from_list(self.list_inference)
         self.exp_name = os.path.basename(self.config_yaml.split(".")[0])
         self.exp_group_name = os.path.basename(os.path.dirname(self.config_yaml))

         if self.reload_from_ckpt is not None:
             self.configs["reload_from_ckpt"] = self.reload_from_ckpt
+        self.dataset_key = build_dataset_json_from_list(self.list_inference)
         self.exp_name = os.path.basename(self.config_yaml.split(".")[0])
         self.exp_group_name = os.path.basename(os.path.dirname(self.config_yaml))