Skip to content

Commit 254e76b

Browse files
committed
Merge remote-tracking branch 'origin/dev' into dev
2 parents 8bb283d + af69ce9 commit 254e76b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

exllamav2/rope.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ def get_rope_params_su(
2323
scaling_factor = math.sqrt(1 + math.log(a / b) / math.log(b))
2424
else:
2525
ext_factors = torch.tensor(cfg.scale_short_factor, dtype = torch.float32, device = device)
26+
scaling_factor = 1.0
2627

2728
inv_freq = 1.0 / (ext_factors * base ** (torch.arange(0, head_dim, 2, device = device).float() / head_dim))
2829
return inv_freq, scaling_factor
@@ -133,6 +134,7 @@ def linear_ramp_factor(min, max, dim):
133134
)
134135
else:
135136
inv_freq = 1.0 / (base ** (torch.arange(0, head_dim, 2, device=device).float() / head_dim))
137+
scaling_factor = 1.0
136138

137139
return inv_freq, scaling_factor
138140

0 commit comments

Comments
 (0)