Guanzheng commited on
Commit
524d0c8
1 Parent(s): 7ca0887

Update clex_layer.py

Browse files
Files changed (1) hide show
  1. clex_layer.py +1 -1
clex_layer.py CHANGED
@@ -60,7 +60,7 @@ class ODELinear(nn.Module):
60
 
61
  class CLEXScalingRotaryEmbedding(nn.Module):
62
 
63
- def __init__(self, dim, max_position_embeddings=2048, rope_scaling=None, base=1000000, device=None) -> None:
64
  super().__init__()
65
 
66
  self.max_t = rope_scaling["max_factor"]
 
60
 
61
  class CLEXScalingRotaryEmbedding(nn.Module):
62
 
63
+ def __init__(self, dim, max_position_embeddings=2048, rope_scaling=None, base=10000, device=None) -> None:
64
  super().__init__()
65
 
66
  self.max_t = rope_scaling["max_factor"]