Spaces:

Icar
/

sdtryout

Runtime error

Icar commited on Sep 9, 2023

Commit

a2d1cee

1 Parent(s): dc1c9c7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 import torch
 import diffusers
-pipe=diffusers.StableDiffusionPipeline.from_pretrained('stabilityai/stable-diffusion-2-1',torch_dtype=torch.float16)
 pipe.to('cuda')
 def gen(prompt):
   img=pipe(prompt)[0][0]

 import gradio as gr
 import torch
 import diffusers
+pipe=diffusers.StableDiffusionPipeline.from_pretrained('stabilityai/stable-diffusion-2-1',torch_dtype=torch.float32)
 pipe.to('cuda')
 def gen(prompt):
   img=pipe(prompt)[0][0]