Spaces:

mygyasir
/

stablediff

Runtime error

mygyasir commited on Aug 2, 2023

Commit

712dadd

•

1 Parent(s): 85da16d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ from diffusers import DiffusionPipeline
 import torch
 import gradio as gr
-pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
 pipe.to("cpu") # Note the change here
 def generate_image(prompt):

 import torch
 import gradio as gr
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float32, use_safetensors=True, variant="fp16")
 pipe.to("cpu") # Note the change here
 def generate_image(prompt):