Skip to content

[plugin] Flux2 model support#606

Draft
Phi-C wants to merge 7 commits intoROCm:mainfrom
Phi-C:flux2
Draft

[plugin] Flux2 model support#606
Phi-C wants to merge 7 commits intoROCm:mainfrom
Phi-C:flux2

Conversation

@Phi-C
Copy link
Copy Markdown

@Phi-C Phi-C commented Apr 19, 2026 •
edited
Loading

This PR add Flux2 model for vLLM-Omni-ATOM, including:

  1. add flux2_trasnformer.py based on tensor-parallel linear layers with ATOM's AITER-backed implementations
  2. update the recipe

Depend on PR 539

tjtanaavllm and others added 6 commits April 19, 2026 11:56
@tjtanaavllm @Phi-C
add vllm omni plugin for QwenImage and Wan22
5f04423 
Signed-off-by: tjtanaavllm <tunjian.tan@amd.com>
@tjtanaavllm @Phi-C
remove qwenimage omni modeling file
a8e8b17 
Signed-off-by: tjtanaavllm <tunjian.tan@amd.com>
@tjtanaavllm @Phi-C
add back qwen image support and add wan22 recipe
70d65bc 
Signed-off-by: tjtanaavllm <tunjian.tan@amd.com>
@tjtanaavllm @Phi-C
fix precommit
0d10675 
Signed-off-by: tjtanaavllm <tunjian.tan@amd.com>
@Phi-C
[plugin] Add Flux2 model support
aca1586
@Phi-C
pix pre-checks format problem
be14816
@Phi-C Phi-C requested a review from zejunchen-zejun April 20, 2026 03:16
@Phi-C Phi-C marked this pull request as draft April 20, 2026 05:45
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@zejunchen-zejun zejunchen-zejun Awaiting requested review from zejunchen-zejun

At least 1 approving review is required to merge this pull request.

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@Phi-C @tjtanaavllm