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