Update clex_layer.py
Browse files- 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=
|
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"]
|