Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
|
|
| 6 |
|
| 7 |
|
| 8 |
flash_pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash").to("cuda", torch.float16)
|
| 9 |
-
flash_pipe.scheduler = EulerDiscreteScheduler.from_config(
|
| 10 |
clip_slider = CLIPSliderXL(flash_pipe, device=torch.device("cuda"))
|
| 11 |
|
| 12 |
@spaces.GPU
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
flash_pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash").to("cuda", torch.float16)
|
| 9 |
+
flash_pipe.scheduler = EulerDiscreteScheduler.from_config(flash_pipe.scheduler.config)
|
| 10 |
clip_slider = CLIPSliderXL(flash_pipe, device=torch.device("cuda"))
|
| 11 |
|
| 12 |
@spaces.GPU
|