AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix ExLlamaV2 loaders using unnecessary "bits" metadata

Browse Source
This commit is contained in:
oobabooga 2024-03-30 21:51:39 -07:00
parent 624faa1438
commit db5f6cd1d8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/models_settings.py
View File

@ -89,7 +89,8 @@ def get_model_metadata(model):
if metadata['rope_scaling']['type'] == 'linear': if metadata['rope_scaling']['type'] == 'linear':
model_settings['compress_pos_emb'] = metadata['rope_scaling']['factor'] model_settings['compress_pos_emb'] = metadata['rope_scaling']['factor']
if 'quantization_config' in metadata: # Read GPTQ metadata for old GPTQ loaders
if 'quantization_config' in metadata and metadata['quantization_config'].get('quant_method', '') != 'exl2':
if 'bits' in metadata['quantization_config']: if 'bits' in metadata['quantization_config']:
model_settings['wbits'] = metadata['quantization_config']['bits'] model_settings['wbits'] = metadata['quantization_config']['bits']
if 'group_size' in metadata['quantization_config']: if 'group_size' in metadata['quantization_config']: