Skip to content

GitLab

You need to sign in or sign up before continuing.
Unverified Commit fffd964a authored by DefTruth's avatar DefTruth Committed by GitHub
Browse files

fix FLUX.2 context parallel (#12737)

parent 859b8090
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/models/transformers/transformer_flux2.py
View file @ fffd964a
......@@ -676,8 +676,8 @@ class Flux2Transformer2DModel(
"": {
"hidden_states": ContextParallelInput(split_dim=1, expected_dims=3, split_output=False),
"encoder_hidden_states": ContextParallelInput(split_dim=1, expected_dims=3, split_output=False),
"img_ids": ContextParallelInput(split_dim=0, expected_dims=2, split_output=False),
"txt_ids": ContextParallelInput(split_dim=0, expected_dims=2, split_output=False),
"img_ids": ContextParallelInput(split_dim=1, expected_dims=3, split_output=False),
"txt_ids": ContextParallelInput(split_dim=1, expected_dims=3, split_output=False),
},
"proj_out": ContextParallelOutput(gather_dim=1, expected_dims=3),
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment