Spaces:
Sleeping
Sleeping
Update clip_slider_pipeline.py
Browse files- 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 |
-
|
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 |
|