SUPIR

Running on Zero

Fabrice-TIERCELIN commited on Apr 19

Commit

68231ac

•

1 Parent(s): 7d92b01

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

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import random
 import imageio
 import torch
-from diffusers import AutoPipelineForImage2Image
 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 = pipe.to(device)
 def update_seed(is_randomize_seed, seed):

 import imageio
 import torch
+from diffusers import DiffusionPipeline
 from PIL import Image, ImageFilter
 max_64_bit_int = 2**63 - 1
 floatType = torch.float16 if torch.cuda.is_available() else torch.float32
 variant = "fp16" if torch.cuda.is_available() else None
+pipe = DiffusionPipeline.from_pretrained("yushan777/SUPIR", trust_remote_code = True)
 pipe = pipe.to(device)
 def update_seed(is_randomize_seed, seed):