Update warp_design_on_dress.py

warp_design_on_dress.py

@@ -26,9 +26,8 @@ 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()
 
-    opt = Options()
-    gmm = GMM(opt)
-    load_checkpoint(gmm, gmm_ckpt, strict = False)
+    gmm = GMM(opt = None)
+    load_checkpoint(gmm, gmm_ckpt)
     gmm.cpu().eval()
 
     with torch.no_grad():