mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add Support for Static NTK RoPE scaling for exllama/exllama_hf (#2955)

Browse source
This commit is contained in:
Panchovix 2023-07-04 00:13:16 -04:00 • committed by GitHub
parent 1610d5ffb2
commit 10c8c197bf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 18 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
modules/ui.py
View file

@ -63,9 +63,11 @@ def list_model_elements():
'llama_cpp_seed',
'gpu_split',
'max_seq_len',
'compress_pos_emb'
'compress_pos_emb',
'alpha_value'
]
for i in range(torch.cuda.device_count()):
elements.append(f'gpu_memory_{i}')