Spaces:

ZhouZJ36DL
/

Multi-turn_Consistent_Image_Editing_FLUX.1-dev

Running on Zero

ZhouZJ36DL commited on May 10

Commit

8e8c8d2

1 Parent(s): df97e04

modified: src/flux/modules/conditioner.py

Files changed (1) hide show

src/flux/modules/conditioner.py CHANGED Viewed

@@ -43,7 +43,7 @@ class HFEmbedder(nn.Module):
         outputs = self.hf_module(
             input_ids=input_ids.to(self.hf_module.device),
-            attention_mask=None,
             output_hidden_states=False,
         )
         return outputs[self.output_key]

         outputs = self.hf_module(
             input_ids=input_ids.to(self.hf_module.device),
+            attention_mask=batch_encoding["attention_mask"].to(self.hf_module.device),
             output_hidden_states=False,
         )
         return outputs[self.output_key]