Spaces:
Running
on
Zero
Running
on
Zero
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"), iterations=
|
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=50)
|
11 |
|
12 |
@spaces.GPU
|
13 |
def generate(slider_x, slider_y, prompt, x_concept_1, x_concept_2, y_concept_1, y_concept_2):
|