linoyts HF staff commited on
Commit
45d6843
1 Parent(s): 39057ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,7 +7,8 @@ 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=100)
 
11
 
12
  @spaces.GPU
13
  def generate(slider_x, slider_y, prompt, x_concept_1, x_concept_2, y_concept_1, y_concept_2, avg_diff_x, avg_diff_y):
 
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, target_word="happy",
11
+ opposite="sad", device=torch.device("cuda"), iterations=100)
12
 
13
  @spaces.GPU
14
  def generate(slider_x, slider_y, prompt, x_concept_1, x_concept_2, y_concept_1, y_concept_2, avg_diff_x, avg_diff_y):