Spaces:

Geek7
/

tadmztxi

Runtime error

Geek7 commited on Oct 6, 2024

Commit

19d367a

verified ·

1 Parent(s): 6c352e2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from diffusers import DiffusionPipeline
 import torch
 # Load the model
-pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v4", torch_dtype=torch.float16).to("cuda")
 def generate_image(prompt):
     image = pipe(prompt).images[0]

 import torch
 # Load the model
+pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v4", torch_dtype=torch.float32).to("cpu")
 def generate_image(prompt):
     image = pipe(prompt).images[0]