Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from diffusers import StableDiffusionXLPipeline, EulerDiscreteScheduler
|
|
| 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
|
| 13 |
def generate(slider_x, slider_y, prompt, x_concept_1, x_concept_2, y_concept_1, y_concept_2):
|
|
|
|
| 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"), iterations=150)
|
| 11 |
|
| 12 |
@spaces.GPU
|
| 13 |
def generate(slider_x, slider_y, prompt, x_concept_1, x_concept_2, y_concept_1, y_concept_2):
|