dreamshaperx-v2-Img2Img-CPU

Runtime error

michaelj commited on Apr 5

Commit

7de84f4

•

1 Parent(s): 774699c

Lykon/dreamshaper-xl-v2-turbo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ from diffusers.utils import load_image
 import math
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = AutoPipelineForImage2Image.from_pretrained("dataautogpt3/ProteusV0.4-Lightning", torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo")
 pipe = pipe.to(device)
 def resize(value,img):

 import math
 device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = AutoPipelineForImage2Image.from_pretrained("Lykon/dreamshaper-xl-v2-turbo", torch_dtype=torch.float16) if torch.cuda.is_available() else AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo")
 pipe = pipe.to(device)
 def resize(value,img):