Spaces:

MUTSC
/

Text-to-Image

Runtime error

MUTSC commited on Apr 15

Commit

56914ca

•

1 Parent(s): e1bcafb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from PIL import Image
 import torch
 # Load the Stable Diffusion model
-model = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)
 model = model.to("cuda" if torch.cuda.is_available() else "cpu")
 def generate_image(prompt):

 import torch
 # Load the Stable Diffusion model
+model = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float32)
 model = model.to("cuda" if torch.cuda.is_available() else "cpu")
 def generate_image(prompt):