Hugging Face's logo

Spaces:
Manjushri
/
OJ-V4-CPU
Runtime error

App Files Community
Manjushri commited on
Commit
0f13a1c
•
1 Parent(s): f07f9a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from PIL import Image
6
  from diffusers import DiffusionPipeline
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
- pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-v2", safety_checker=None)
10
  pipe = pipe.to(device)
11
 
12
  def genie (prompt, scale, steps, seed):
 
6
  from diffusers import DiffusionPipeline
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
+ pipe = DiffusionPipeline.from_pretrained("prompthero/openjourney-lora", safety_checker=None)
10
  pipe = pipe.to(device)
11
 
12
  def genie (prompt, scale, steps, seed):