Time-Stream

Paused

linoyts HF Staff commited on Jul 26, 2024

Commit

41667f9

verified ·

1 Parent(s): ecc78e5

Update clip_slider_pipeline.py

Files changed (1) hide show

clip_slider_pipeline.py CHANGED Viewed

@@ -239,7 +239,7 @@ class CLIPSliderXL(CLIPSlider):
                     toks.to(text_encoder.device),
                     output_hidden_states=True,
                 )
                 # We are only ALWAYS interested in the pooled output of the final text encoder
                 pooled_prompt_embeds = prompt_embeds[0]
                 prompt_embeds = prompt_embeds.hidden_states[-2]

                     toks.to(text_encoder.device),
                     output_hidden_states=True,
                 )
+                print("prompt_embeds.dtype",prompt_embeds.dtype)
                 # We are only ALWAYS interested in the pooled output of the final text encoder
                 pooled_prompt_embeds = prompt_embeds[0]
                 prompt_embeds = prompt_embeds.hidden_states[-2]