[diffusion] fix: enable torch.compile for UlyssesAttention (#18840)

2026-02-15 13:24:27 +05:30
parent ad1bdb93df
commit 274bf6607a
1 changed files with 0 additions and 2 deletions
--- a/python/sglang/multimodal_gen/runtime/layers/attention/layer.py
+++ b/python/sglang/multimodal_gen/runtime/layers/attention/layer.py
@@ -78,7 +78,6 @@ class UlyssesAttention(nn.Module):
        self.backend = attn_backend.get_enum()
        self.dtype = dtype

-    @torch.compiler.disable
    def forward(
        self,
        q: torch.Tensor,
@@ -157,7 +156,6 @@ class UlyssesAttention(nn.Module):
 class UlyssesAttention_VSA(UlyssesAttention):
    """Distributed attention layer with VSA support."""

-    @torch.compiler.disable
    def forward(
        self,
        q: torch.Tensor,