Spaces:

Manjushri
/

Manju-Dream-Booth-GPU

Running on T4

Manjushri commited on Mar 24, 2023

Commit

c24bef3

•

1 Parent(s): d83e9f0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from PIL import Image
 from diffusers import DiffusionPipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v2", torch_dtype=torch.float16, revision="fp16", safety_checker=None)
 pipe = pipe.to(device)
 def genie (prompt, scale, steps, seed):

 from diffusers import DiffusionPipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v2", torch_dtype=torch.float16, safety_checker=None)
 pipe = pipe.to(device)
 def genie (prompt, scale, steps, seed):