Update model/pipeline.py

model/pipeline.py

@@ -34,7 +34,7 @@ class CatVTONPipeline:
     ):
         self.device = device
         self.weight_dtype = weight_dtype
-        self.skip_safety_check = skip_safety_check
+        self.skip_safety_check = True
 
         self.noise_scheduler = DDIMScheduler.from_pretrained(base_ckpt, subfolder="scheduler")
         self.vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(device, dtype=weight_dtype)