Skip to content

[Diffusion] Enable vLLM-Omni Plugin for Diffusion Model#1

Draft
tjtanaa wants to merge 7 commits intomainfrom
vllmomni
Draft

[Diffusion] Enable vLLM-Omni Plugin for Diffusion Model#1
tjtanaa wants to merge 7 commits intomainfrom
vllmomni

Conversation

@tjtanaa
Copy link
Copy Markdown
Member

@tjtanaa tjtanaa commented Mar 31, 2026

No description provided.

tjtanaa added 7 commits March 16, 2026 08:42
Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>
Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>
Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>
Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>
Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>
Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com>
Comment on lines +5 to +6
import torch.nn.functional as F
from vllm.logger import init_logger
Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

⚠️ [ruff] <F401> reported by reviewdog 🐶
torch.nn.functional imported but unused

Suggested change
import torch.nn.functional as F
from vllm.logger import init_logger
from vllm.logger import init_logger

Comment on lines +11 to +19
from vllm_omni.diffusion.models.wan2_2.wan2_2_transformer import (
ColumnParallelGELU,
WanCrossAttention,
WanFeedForward,
WanSelfAttention,
WanTransformerBlock,
WanTransformer3DModel,
apply_rotary_emb_wan,
)
Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

⚠️ [ruff] <F401> reported by reviewdog 🐶
vllm_omni.diffusion.models.wan2_2.wan2_2_transformer.ColumnParallelGELU imported but unused

Suggested change
from vllm_omni.diffusion.models.wan2_2.wan2_2_transformer import (
ColumnParallelGELU,
WanCrossAttention,
WanFeedForward,
WanSelfAttention,
WanTransformerBlock,
WanTransformer3DModel,
apply_rotary_emb_wan,
)
from vllm_omni.diffusion.models.wan2_2.wan2_2_transformer import (
WanCrossAttention,
WanFeedForward,
WanSelfAttention,
WanTransformerBlock,
WanTransformer3DModel,
apply_rotary_emb_wan,
)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant