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)
|