Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from diffusers import StableDiffusionXLPipeline, EDMEulerScheduler
+from diffusers import StableDiffusionXLPipeline, EDMEulerScheduler, StableDiffusionXLInstructPix2PixPipeline
 from custom_pipeline import CosStableDiffusionXLInstructPix2PixPipeline
 from huggingface_hub import hf_hub_download
 import numpy as np
@@ -40,8 +40,8 @@ def resize_image(image, resolution):
 
 EDMEulerScheduler.set_timesteps = set_timesteps_patched
 
-pipe_edit = CosStableDiffusionXLInstructPix2PixPipeline.from_single_file(
-    edit_file, num_in_channels=8
+pipe_edit = StableDiffusionXLInstructPix2PixPipeline.from_single_file(
+    edit_file, num_in_channels=8, is_cosxl_edit=True, torch_dtype=torch.float16
 )
 pipe_edit.scheduler = EDMEulerScheduler(sigma_min=0.002, sigma_max=120.0, sigma_data=1.0, prediction_type="v_prediction")
 pipe_edit.to("cuda")