Update CatVTON_model.py

CatVTON_model.py

@@ -16,7 +16,7 @@ class CatVTONPix2PixPipeline:
     def __init__(
         self, 
         weight_dtype=torch.float32,
-        device='cuda',
+        device='cpu',
         compile=False,
         skip_safety_check=True,
         use_tf32=True,
@@ -35,9 +35,9 @@ class CatVTONPix2PixPipeline:
  
         self.unet=models.get('diffusion', None)  
         # # Enable TF32 for faster training on Ampere GPUs (A100 and RTX 30 series).
-        if use_tf32:
-            torch.set_float32_matmul_precision("high")
-            torch.backends.cuda.matmul.allow_tf32 = True
+        # if use_tf32:
+        #     torch.set_float32_matmul_precision("high")
+        #     torch.backends.cuda.matmul.allow_tf32 = True
 
     @torch.no_grad()
     def __call__(