Remove print statement in DeepseekScalingRotaryEmbedding (#11604)

2025-10-20 23:03:52 +08:00 · 2024-12-29 15:16:46 -05:00
parent faef77c0d6
commit 0aa38d16f5
1 changed files with 0 additions and 1 deletions
--- a/vllm/model_executor/layers/rotary_embedding.py
+++ b/vllm/model_executor/layers/rotary_embedding.py
@ -668,7 +668,6 @@ class DeepseekScalingRotaryEmbedding(RotaryEmbedding):
        cos = (freqs.cos() * self.mscale)
        sin = (freqs.sin() * self.mscale)
        cache = torch.cat((cos, sin), dim=-1)
-        print("Cache shape", cache.shape)
        return cache

    def forward(