Skip to content

GitLab

Unverified Commit 0aa38d16 authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

Remove print statement in DeepseekScalingRotaryEmbedding (#11604)

parent faef77c0
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
vllm/model_executor/layers/rotary_embedding.py
View file @ 0aa38d16
......@@ -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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment