Update README.md

README.md

@@ -58,7 +58,7 @@ original_image = original_image.resize((768, 512))
 from diffusers import AutoPipelineForImage2Image
 import torch
 
-pipe = AutoPipelineForImage2Image.from_pretrained("kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float16)
+pipe = AutoPipelineForImage2Image.from_pretrained("Shaleen123/kandinsky_2.5", torch_dtype=torch.float16)
 pipe.enable_model_cpu_offload()
 
 prompt = "A fantasy landscape, Cinematic lighting"
@@ -104,7 +104,7 @@ weights = [0.3, 0.3, 0.4]
 prompt = ""
 prior_out = pipe_prior.interpolate(images_texts, weights)
 
-pipe = KandinskyV22Pipeline.from_pretrained("kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float16)
+pipe = KandinskyV22Pipeline.from_pretrained("Shaleen123/kandinsky_2.5", torch_dtype=torch.float16)
 pipe.to("cuda")
 
 image = pipe(**prior_out, height=768, width=768).images[0]