update

options/Banner_Model/Image2Image_2.py

@@ -18,7 +18,7 @@ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 # pipe.enable_model_cpu_offload()
 
 def I2I_2(image, prompt,size,num_inference_steps):
-    image.resize((size))
+    image.resize((size,size))
     image=processor(image)
     generator = torch.Generator(device=device).manual_seed(0)
     image = pipe(prompt, num_inference_steps=num_inference_steps, generator=generator, image=image).images[0]