Update warp_design_on_dress.py

warp_design_on_dress.py

@@ -3,7 +3,7 @@ import torch
 import torch.nn.functional as F
 from torchvision import transforms
 from PIL import Image
-from networks import GMM, UnetGenerator, load_checkpoint
+from networks import GMM, UnetGenerator, load_checkpoint, Options
 
 
 def run_design_warp_on_dress(dress_path, design_path, gmm_ckpt, tom_ckpt, output_dir):
@@ -25,9 +25,10 @@ def run_design_warp_on_dress(dress_path, design_path, gmm_ckpt, tom_ckpt, output
 
     # Fake agnostic: use the dress image itself
     agnostic = dress_tensor.clone()
-
-    gmm = GMM(opt = None)
-    load_checkpoint(gmm, gmm_ckpt)
+    
+    opt = Options()
+    gmm = GMM(opt)
+    load_checkpoint(gmm, gmm_ckpt, strict = False)
     gmm.cpu().eval()
 
     with torch.no_grad():