Update app.py

app.py

@@ -34,7 +34,7 @@ def pipe_t2i():
     return PIPE
     
 def pipe_i2i():
-    PIPE = DiffusionPipeline.from_pretrained("yisol/IDM-VTON", torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to("cuda")
+    PIPE = DiffusionPipeline.from_pretrained("yisol/IDM-VTON", torch_dtype=torch.float16).to("cuda")
     PIPE.unet = torch.compile(PIPE.unet, mode="reduce-overhead", fullgraph=True)
     return PIPE