Update warp_design_on_dress.py

warp_design_on_dress.py

@@ -26,10 +26,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 = GMM(opt=None)
-    load_checkpoint(gmm, gmm_ckpt)
-    gmm.cuda().eval()
+    opt = Options()
+    gmm = GMM(opt)
+    load_checkpoint(gmm, gmm_ckpt, strict = False)
+    gmm.cpu().eval()
 
     with torch.no_grad():
         grid, _ = gmm(agnostic, design_mask)