Update app.py

app.py

@@ -4,7 +4,8 @@ import numpy as np
 import random
 import spaces
 import torch
-from diffusers import DiffusionPipeline
+#from diffusers import DiffusionPipeline
+from diffusers import AutoPipelineForImage2Image
 from huggingface_hub import InferenceClient
 
 dtype = torch.bfloat16
@@ -17,7 +18,9 @@ print('sdxl loaded')
 
 "kandinsky-community/kandinsky-2-2-decoder"
 #pipeline2Image = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtypes=torch.bfloat16).to(device)
-pipeline2Image = DiffusionPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-decoder", torch_dtypes=torch.bfloat16).to(device)
+#pipeline2Image = DiffusionPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-decoder", torch_dtypes=torch.bfloat16).to(device)
+pipeline2Image = AutoPipelineForImage2Image.from_pretrained("kandinsky-community/kandinsky-2-2-decoder", torch_dtype=dtype)
+pipeline2Image.enable_model_cpu_offload()
 print("pipeline 2 image loaded")
 
 MAX_SEED = np.iinfo(np.int32).max