mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-09-20 10:35:10 +02:00
Add missing file
This commit is contained in:
parent
f4f04c8c32
commit
21058c37f7
1 changed files with 18 additions and 0 deletions
18
modules/RoPE.py
Normal file
18
modules/RoPE.py
Normal file
|
@ -0,0 +1,18 @@
|
|||
def get_alpha_value(alpha, base):
|
||||
'''
|
||||
Gets alpha_value from alpha_value and rope_freq_base
|
||||
'''
|
||||
if base > 0:
|
||||
return (base/10000.) ** (63/64.)
|
||||
else:
|
||||
return alpha
|
||||
|
||||
|
||||
def get_rope_freq_base(alpha, base):
|
||||
'''
|
||||
Gets rope_freq_base from alpha_value and rope_freq_base
|
||||
'''
|
||||
if base > 0:
|
||||
return base
|
||||
else:
|
||||
return 10000 * alpha ** (64/63.)
|
Loading…
Reference in a new issue