oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-04-06 15:13:38 +00:00
Code Issues Projects Releases Packages Wiki Activity

Remove obsolete pre_layer parameter

Browse source
This commit is contained in:
oobabooga 2024-06-12 18:56:44 -07:00
parent 46174a2d33
commit 2d196ed2fe
3 changed files with 1 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
modules/models_settings.py
View file

@ -204,14 +204,11 @@ def update_model_parameters(state, initial=False):
value = vars(shared.args_defaults)[element]
# Making some simple conversions
if element in ['wbits', 'groupsize', 'pre_layer']:
if element in ['wbits', 'groupsize']:
value = int(value)
elif element == 'cpu_memory' and value is not None:
value = f"{value}MiB"
if element in ['pre_layer']:
value = [value] if value > 0 else None
setattr(shared.args, element, value)
found_positive = False