modified: src/flux/modules/conditioner.py

src/flux/modules/conditioner.py

@@ -115,14 +115,10 @@ class HFEmbedder(nn.Module):
             max_pos_emb = getattr(self.hf_module.config, 'max_position_embeddings', -1)
             if max_pos_emb > 0 and position_ids.max() >= max_pos_emb:
                  raise ValueError(f"Generated position_ids max ({position_ids.max().item()}) >= model's max_position_embeddings ({max_pos_emb})")
-
+            # Pass the explicitly created position_ids to the model
+            model_kwargs["position_ids"] = position_ids
         try:
-            outputs = self.hf_module(
-                input_ids=input_ids,
-                attention_mask=attention_mask,
-                output_hidden_states=False,
-                position_ids=position_ids
-            )
+            outputs = self.hf_module(**model_kwargs)
         except IndexError as e:
             # 捕获并提供更详细的错误上下文
             print(f"*** IndexError caught during model forward pass ***")