Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
import torch
|
2 |
-
from diffusers import StableDiffusion3Pipeline, StableDiffusionPipeline,
|
3 |
import gradio as gr
|
4 |
import os
|
5 |
import random
|
@@ -29,7 +29,7 @@ sd2_1_pipe = StableDiffusionPipeline.from_pretrained(
|
|
29 |
)
|
30 |
sd2_1_pipe.to(device)
|
31 |
|
32 |
-
sdxl_pipe =
|
33 |
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
|
34 |
)
|
35 |
sdxl_pipe.to(device)
|
|
|
1 |
import torch
|
2 |
+
from diffusers import StableDiffusion3Pipeline, StableDiffusionPipeline, StableDiffusionXLPipeline, DPMSolverSinglestepScheduler
|
3 |
import gradio as gr
|
4 |
import os
|
5 |
import random
|
|
|
29 |
)
|
30 |
sd2_1_pipe.to(device)
|
31 |
|
32 |
+
sdxl_pipe = StableDiffusionXLPipeline.from_pretrained(
|
33 |
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
|
34 |
)
|
35 |
sdxl_pipe.to(device)
|