Update infer/infer_mos5.py

infer/infer_mos5.py

@@ -12,17 +12,10 @@ import sys
 
 from qa_mdt.audioldm_train.utilities.data.dataset_original_mos5 import AudioDataset as AudioDataset
 from qa_mdt.audioldm_train.utilities.tools import build_dataset_json_from_list
+from qa_mdt.audioldm_train.utilities.model_util import instantiate_from_config
 from torch.utils.data import DataLoader
 from pytorch_lightning import seed_everything
 from qa_mdt.audioldm_train.utilities.tools import get_restore_step
-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):