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
Hide 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): ...@@ -668,7 +668,6 @@ class DeepseekScalingRotaryEmbedding(RotaryEmbedding):
cos = (freqs.cos() * self.mscale) cos = (freqs.cos() * self.mscale)
sin = (freqs.sin() * self.mscale) sin = (freqs.sin() * self.mscale)
cache = torch.cat((cos, sin), dim=-1) cache = torch.cat((cos, sin), dim=-1)
print("Cache shape", cache.shape)
return cache return cache
def forward( 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