oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-03-04 20:53:50 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix the transformers loader

Browse source
This commit is contained in:
oobabooga 2025-04-21 18:33:14 -07:00
parent 8320190184
commit 78aeabca89
1 changed files with 1 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
modules/models_settings.py
View file

@ -197,13 +197,9 @@ def update_model_parameters(state, initial=False):
if initial and element in shared.provided_arguments:
continue
if element in ['cpu_memory'] and value == 0:
if element == 'cpu_memory' and value == 0:
value = vars(shared.args_defaults)[element]
# Making some simple conversions
if element == 'cpu_memory' and value is not None:
value = f"{value}MiB"
setattr(shared.args, element, value)