Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -6,12 +6,10 @@ from safetensors.torch import load_file
|
|
6 |
import spaces
|
7 |
|
8 |
### SDXL Turbo ####
|
9 |
-
|
10 |
pipe_turbo = StableDiffusionXLPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
|
11 |
pipe_turbo.to("cuda")
|
12 |
|
13 |
### SDXL Lightning ###
|
14 |
-
|
15 |
base = "stabilityai/stable-diffusion-xl-base-1.0"
|
16 |
repo = "ByteDance/SDXL-Lightning"
|
17 |
ckpt = "sdxl_lightning_1step_unet_x0.safetensors"
|
|
|
6 |
import spaces
|
7 |
|
8 |
### SDXL Turbo ####
|
|
|
9 |
pipe_turbo = StableDiffusionXLPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
|
10 |
pipe_turbo.to("cuda")
|
11 |
|
12 |
### SDXL Lightning ###
|
|
|
13 |
base = "stabilityai/stable-diffusion-xl-base-1.0"
|
14 |
repo = "ByteDance/SDXL-Lightning"
|
15 |
ckpt = "sdxl_lightning_1step_unet_x0.safetensors"
|