From 78aeabca89275939f66c58dff7530966d9886a18 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Mon, 21 Apr 2025 18:33:14 -0700 Subject: [PATCH] Fix the transformers loader --- modules/models_settings.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/modules/models_settings.py b/modules/models_settings.py index f5f3be7a..ee2ed71b 100644 --- a/modules/models_settings.py +++ b/modules/models_settings.py @@ -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)