Remove breakable_cudagraph import (not in nightly)

This commit is contained in:
2026-05-18 22:29:24 +00:00
parent b007937a68
commit a19ed4a18e

View File

@@ -14,7 +14,6 @@ import torch.nn.functional as F
from transformers import DeepseekV2Config, DeepseekV3Config
import vllm.envs as envs
from vllm.compilation.breakable_cudagraph import eager_break_during_capture
from vllm.model_executor.layers.linear import (
ReplicatedLinear,
)
@@ -554,7 +553,6 @@ class DeepseekV4MultiHeadLatentAttentionWrapper(PluggableLayer):
)
@eager_break_during_capture
def deepseek_v4_attention(
hidden_states: torch.Tensor,
positions: torch.Tensor,