linoyts HF staff commited on
Commit
e0bab07
1 Parent(s): 7ec377d

Update app.py (#3)

Browse files

- Update app.py (2333d47c0f86ce3dd007624f982f6378f20f77f7)
- Update clip_slider_pipeline.py (53c6db545a065ecb5b3c71809264664da4055fce)

Files changed (2) hide show
  1. app.py +2 -2
  2. clip_slider_pipeline.py +0 -3
app.py CHANGED
@@ -69,8 +69,8 @@ def generate(prompt,
69
  ):
70
  slider_x = [concept_2, concept_1]
71
  # check if avg diff for directions need to be re-calculated
72
- print("slider_x", slider_x)
73
- print("x_concept_1", x_concept_1, "x_concept_2", x_concept_2)
74
  if randomize_seed:
75
  seed = random.randint(0, MAX_SEED)
76
 
 
69
  ):
70
  slider_x = [concept_2, concept_1]
71
  # check if avg diff for directions need to be re-calculated
72
+ #print("slider_x", slider_x)
73
+ #print("x_concept_1", x_concept_1, "x_concept_2", x_concept_2)
74
  if randomize_seed:
75
  seed = random.randint(0, MAX_SEED)
76
 
clip_slider_pipeline.py CHANGED
@@ -412,7 +412,6 @@ class CLIPSliderFlux(CLIPSlider):
412
  # if pooler token only [-4,4] work well
413
 
414
  with torch.no_grad():
415
- print("WTF 1.25")
416
  text_inputs = self.pipe.tokenizer(
417
  prompt,
418
  padding="max_length",
@@ -454,11 +453,9 @@ class CLIPSliderFlux(CLIPSlider):
454
  if avg_diff_2nd is not None:
455
  pooled_prompt_embeds += avg_diff_2nd * scale_2nd
456
 
457
- print("WTF 1.5")
458
  torch.manual_seed(seed)
459
  images = self.pipe(prompt_embeds=prompt_embeds, pooled_prompt_embeds=pooled_prompt_embeds,
460
  **pipeline_kwargs).images
461
- print("WTF 1.75")
462
 
463
  return images[0]
464
 
 
412
  # if pooler token only [-4,4] work well
413
 
414
  with torch.no_grad():
 
415
  text_inputs = self.pipe.tokenizer(
416
  prompt,
417
  padding="max_length",
 
453
  if avg_diff_2nd is not None:
454
  pooled_prompt_embeds += avg_diff_2nd * scale_2nd
455
 
 
456
  torch.manual_seed(seed)
457
  images = self.pipe(prompt_embeds=prompt_embeds, pooled_prompt_embeds=pooled_prompt_embeds,
458
  **pipeline_kwargs).images
 
459
 
460
  return images[0]
461