modify

app.py

@@ -30,7 +30,7 @@ from tqdm import tqdm
 os.makedirs('models', exist_ok=True)
 model_path = "models/relvid_mm_sd15_fbc_unet.pth"
 
-if not os.path.exists(filename):
+if not os.path.exists(model_path):
     download_url_to_file(url='https://huggingface.co/aleafy/RelightVid/resolve/main/relvid_mm_sd15_fbc_unet.pth', dst=model_path)
 
 # if not os.path.exists(filename):

misc_utils/train_utils.py

@@ -29,13 +29,10 @@ def get_models(args):
 
 def get_text_model(args):
     base_path = None
-    train_net = None
     if args.get('diffusion'):
         if args.diffusion.params.get('base_path'):# 这边有base path的情况下已经load参数了
             base_path = args.diffusion.params.base_path
-            train_net = args.diffusion.params.get('unet_init_weights')
     if args.get('text_model'):
-        args.text_model.params.base_path = base_path
         text_model = instantiate_from_config(args.text_model)
         return text_model
     return None