From 274bf6607a1ff183caeb70b1753823debd98b8cd Mon Sep 17 00:00:00 2001
From: Ratish P <114130421+Ratish1@users.noreply.github.com>
Date: Sun, 15 Feb 2026 13:24:27 +0530
Subject: [PATCH] [diffusion] fix: enable torch.compile for UlyssesAttention
 (#18840)

---
 python/sglang/multimodal_gen/runtime/layers/attention/layer.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/python/sglang/multimodal_gen/runtime/layers/attention/layer.py b/python/sglang/multimodal_gen/runtime/layers/attention/layer.py
index 26bcbdd48..8a10b18a9 100644
--- 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,