Update warp_design_on_dress.py

warp_design_on_dress.py

@@ -33,8 +33,8 @@ def run_design_warp_on_dress(dress_path, design_path, gmm_ckpt, tom_ckpt, output
     gmm.cpu().eval()
 
     # Convert agnostic to 22 channels before passing to GMM
-    agnostic_22ch = pad_to_22_channels(agnostic)
-    design_mask_22ch = pad_to_22_channels(design_mask)
+    agnostic_22ch = pad_to_22_channels(agnostic).contiguous()
+    design_mask_22ch = pad_to_22_channels(design_mask).contiguous()
 
     with torch.no_grad():
         grid, _ = gmm(agnostic_22ch, design_mask_22ch)  # Use padded inputs