DiffusionPipeline.from_pretrained("yushan777/SUPIR", trust_remote_code = True)

app.py

@@ -6,7 +6,7 @@ import random
 import imageio
 import torch
 
-from diffusers import AutoPipelineForImage2Image
+from diffusers import DiffusionPipeline
 from PIL import Image, ImageFilter
 
 max_64_bit_int = 2**63 - 1
@@ -15,7 +15,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 floatType = torch.float16 if torch.cuda.is_available() else torch.float32
 variant = "fp16" if torch.cuda.is_available() else None
 
-pipe = AutoPipelineForImage2Image.from_pretrained("Fabrice-TIERCELIN/sdxl-turbo", torch_dtype = floatType, variant = variant)
+pipe = DiffusionPipeline.from_pretrained("yushan777/SUPIR", trust_remote_code = True)
 pipe = pipe.to(device)
 
 def update_seed(is_randomize_seed, seed):