mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-31 06:30:15 +01:00
19 lines
399 B
Python
19 lines
399 B
Python
|
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.)
|