latentnavigation-flux

Sleeping

linoyts HF staff commited on Jul 26

Commit

a0503f3

•

1 Parent(s): 2f67a39

Update clip_slider_pipeline.py

Files changed (1) hide show

clip_slider_pipeline.py CHANGED Viewed

@@ -236,7 +236,7 @@ class CLIPSliderXL(CLIPSlider):
                 toks = text_inputs.input_ids
                 prompt_embeds = text_encoder(
-                    toks.to(text_encoder.device),
                     output_hidden_states=True,
                 )

                 toks = text_inputs.input_ids
                 prompt_embeds = text_encoder(
+                    toks.to(text_encoder.device, torch.float16),
                     output_hidden_states=True,
                 )