Procházet zdrojové kódy

Update utils_llama.py

Allen před 1 rokem
rodič
revize
5affb02787
1 změnil soubory, kde provedl 0 přidání a 1 odebrání
  1. 0 1
      research/long-context-llama/H2O/utils_llama.py

+ 0 - 1
research/long-context-llama/H2O/utils_llama.py

@@ -43,7 +43,6 @@ def apply_rotary_pos_emb_single(x, cos, sin, position_ids=None, unsqueeze_dim=1)
 
     cos = cos.unsqueeze(unsqueeze_dim)
     sin = sin.unsqueeze(unsqueeze_dim)
-    print(cos.shape, sin.shape, x.shape)
     x_embed = (x * cos) + (rotate_half(x) * sin)
 
     return x_embed