From 2f5d769a8d5eda36b06ff22148bfdc13554d4d8a Mon Sep 17 00:00:00 2001 From: Cebtenzzre Date: Sun, 27 Aug 2023 17:54:43 -0400 Subject: [PATCH] accept floating-point alpha value on the command line (#3712) --- modules/shared.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/shared.py b/modules/shared.py index ca68c1a6..08e7a881 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -159,7 +159,7 @@ parser.add_argument('--rwkv-strategy', type=str, default=None, help='RWKV: The s parser.add_argument('--rwkv-cuda-on', action='store_true', help='RWKV: Compile the CUDA kernel for better performance.') # RoPE -parser.add_argument('--alpha_value', type=int, default=1, help="Positional embeddings alpha factor for NTK RoPE scaling. Use either this or compress_pos_emb, not both.") +parser.add_argument('--alpha_value', type=float, default=1, help="Positional embeddings alpha factor for NTK RoPE scaling. Use either this or compress_pos_emb, not both.") parser.add_argument('--rope_freq_base', type=int, default=0, help="If greater than 0, will be used instead of alpha_value. Those two are related by rope_freq_base = 10000 * alpha_value ^ (64 / 63).") parser.add_argument('--compress_pos_emb', type=int, default=1, help="Positional embeddings compression factor. Should be set to (context length) / (model\'s original context length). Equal to 1/rope_freq_scale.")