linoyts HF staff commited on
Commit
62a0cc0
1 Parent(s): 64b9ad0

Update clip_slider_pipeline.py

Browse files
Files changed (1) hide show
  1. clip_slider_pipeline.py +1 -1
clip_slider_pipeline.py CHANGED
@@ -245,7 +245,7 @@ class CLIPSliderXL(CLIPSlider):
245
  if avg_diff_2nd:
246
  prompt_embeds[:, toks.argmax()] += avg_diff_2nd[0] * scale_2nd
247
  else:
248
- print(avg_diff)
249
  normed_prompt_embeds = prompt_embeds / prompt_embeds.norm(dim=-1, keepdim=True)
250
  sims = normed_prompt_embeds[0] @ normed_prompt_embeds[0].T
251
 
 
245
  if avg_diff_2nd:
246
  prompt_embeds[:, toks.argmax()] += avg_diff_2nd[0] * scale_2nd
247
  else:
248
+
249
  normed_prompt_embeds = prompt_embeds / prompt_embeds.norm(dim=-1, keepdim=True)
250
  sims = normed_prompt_embeds[0] @ normed_prompt_embeds[0].T
251