Remove print statement in DeepseekScalingRotaryEmbedding (vllm-projec…

…t#11604) Signed-off-by: Bowen Wang <abmfy@icloud.com>
abmfy · Jan 24, 2025 · 8abe395 · 8abe395
1 parent cdd3c57
commit 8abe395
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/vllm/model_executor/layers/rotary_embedding.py b/vllm/model_executor/layers/rotary_embedding.py
@@ -668,7 +668,6 @@ def _compute_cos_sin_cache(self) -> torch.Tensor:
         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(