Update warp_design_on_dress.py

warp_design_on_dress.py

@@ -39,7 +39,7 @@ def run_design_warp_on_dress(dress_path, design_path, gmm_ckpt, tom_ckpt, output
     # ----- TOM -----
     tom = UnetGenerator(26, 4, 6, ngf=64, norm_layer=torch.nn.InstanceNorm2d)
     load_checkpoint(tom, tom_ckpt)
-    tom.cuda().eval()
+    tom.cpu().eval()
 
     with torch.no_grad():
         tom_input = torch.cat([agnostic, warped_design, warped_mask], 1)