Merge pull request #21 from haoheliu/haoheliu-patch-1

Update latent_diffusion.py
haoheliu · Nov 26, 2023 · a6b15e8 · a6b15e8
2 parents 58cd823 + 279b604
commit a6b15e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/audioldm_train/train/latent_diffusion.py b/audioldm_train/train/latent_diffusion.py
@@ -238,7 +238,7 @@ def main(configs, config_yaml_path, exp_group_name, exp_name, perform_validation
     config_yaml_path = os.path.join(config_yaml)
     config_yaml = yaml.load(open(config_yaml_path, "r"), Loader=yaml.FullLoader)
 
-    if "reload_from_ckpt" is not None:
+    if args.reload_from_ckpt is not None:
         config_yaml["reload_from_ckpt"] = args.reload_from_ckpt
 
     if perform_validation: