Update infer/infer_mos5.py

infer/infer_mos5.py

@@ -15,7 +15,14 @@ from .audioldm_train.utilities.tools import build_dataset_json_from_list
 from torch.utils.data import DataLoader
 from pytorch_lightning import seed_everything
 from .audioldm_train.utilities.tools import get_restore_step
-from .audioldm_train.utilities.model_util import instantiate_from_config
+def instantiate_from_config(config):
+    if not "target" in config:
+        if config == "__is_first_stage__":
+            return None
+        elif config == "__is_unconditional__":
+            return None
+        raise KeyError("Expected key `target` to instantiate.")
+    return get_obj_from_str(config["target"])(**config.get("params", dict()))
 
 
 def infer(dataset_key, configs, config_yaml_path, exp_group_name, exp_name):