Spaces:
Running
on
Zero
Running
on
Zero
Update clip_slider_pipeline.py
Browse files- clip_slider_pipeline.py +3 -0
clip_slider_pipeline.py
CHANGED
@@ -412,6 +412,7 @@ class CLIPSliderFlux(CLIPSlider):
|
|
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,9 +454,11 @@ class CLIPSliderFlux(CLIPSlider):
|
|
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 |
|
|
|
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 |
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 |
|