linoyts HF staff commited on
Commit
78dd4cd
1 Parent(s): 7f1272f

Update clip_slider_pipeline.py

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