Remove print statement in DeepseekScalingRotaryEmbedding (vllm-projec…

…t#11604) Signed-off-by: xcnick <[email protected]>
xcnick · Dec 31, 2024 · a54cf6c · a54cf6c
1 parent 41b22e3
commit a54cf6c
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(