Remove print statement in DeepseekScalingRotaryEmbedding (vllm-projec…

…t#11604)
BKitor · Dec 30, 2024 · 3acd6a1 · 3acd6a1
1 parent 6b8ce76
commit 3acd6a1
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(