SUPIR

Running

Fabrice-TIERCELIN commited on Apr 19

Commit

7d92b01

•

1 Parent(s): 36a8419

Fabrice-TIERCELIN/sdxl-turbo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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("stabilityai/sdxl-turbo", torch_dtype = floatType, variant = variant)
 pipe = pipe.to(device)
 def update_seed(is_randomize_seed, seed):

 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):