Spaces:
Runtime error
Runtime error
Commit
·
42cd0c7
1
Parent(s):
b29e29f
try diffusion fix
Browse files
app.py
CHANGED
@@ -5,8 +5,9 @@ from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
|
|
5 |
import torch
|
6 |
|
7 |
model_id = "stabilityai/stable-diffusion-2-1"
|
8 |
-
pipe = StableDiffusionPipeline.from_pretrained(
|
9 |
-
pipe
|
|
|
10 |
# pipe = pipe.to("cuda")
|
11 |
|
12 |
openai.api_key = os.getenv("OPENAI_API_KEY")
|
|
|
5 |
import torch
|
6 |
|
7 |
model_id = "stabilityai/stable-diffusion-2-1"
|
8 |
+
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
|
9 |
+
# pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
10 |
+
# pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
11 |
# pipe = pipe.to("cuda")
|
12 |
|
13 |
openai.api_key = os.getenv("OPENAI_API_KEY")
|