increase speed (test)

app.py

@@ -7,10 +7,8 @@ import numpy as np
 import torch
 from PIL import Image
 from diffusers import DiffusionPipeline, StableDiffusionXLPipeline, EDMEulerScheduler, StableDiffusionXLInstructPix2PixPipeline, AutoencoderKL
-from custom_pipeline import CosStableDiffusionXLInstructPix2PixPipeline
 from huggingface_hub import hf_hub_download
 from huggingface_hub import InferenceClient
-from diffusers import StableDiffusion3Pipeline, SD3Transformer2DModel, FlowMatchEulerDiscreteScheduler
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16
@@ -51,9 +49,7 @@ def set_timesteps_patched(self, num_inference_steps: int, device = None):
 # Image Editor
 edit_file = hf_hub_download(repo_id="stabilityai/cosxl", filename="cosxl_edit.safetensors")
 EDMEulerScheduler.set_timesteps = set_timesteps_patched
-pipe_edit = StableDiffusionXLInstructPix2PixPipeline.from_single_file(
-    edit_file, num_in_channels=8, is_cosxl_edit=True, vae=vae, torch_dtype=torch.float16,
-)
+pipe_edit = StableDiffusionXLInstructPix2PixPipeline.from_single_file( edit_file, num_in_channels=8, is_cosxl_edit=True, vae=vae, 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")
 
@@ -103,8 +99,8 @@ def king(type ,
             prompt = instruction,
             guidance_scale = guidance_scale, 
             num_inference_steps = steps, 
-            width = width, 
-            height = height,
+            width = (width/2), 
+            height = (height/2),
             generator = generator,
             output_type="latent",
         ).images         
@@ -113,6 +109,8 @@ def king(type ,
             prompt=instruction,
             guidance_scale=guidance_scale,
             num_inference_steps=steps,
+            width = width, 
+            height = height,
             image=image,
             generator=generator,
         ).images[0]