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

Style changes

Browse source
This commit is contained in:
oobabooga 2023-11-15 15:51:37 -08:00
parent be125e2708
commit e05d8fd441
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
modules/models_settings.py
View file

@ -127,8 +127,10 @@ def infer_loader(model_name, model_settings):
return loader return loader
# UI: update the command-line arguments based on the interface values
def update_model_parameters(state, initial=False): def update_model_parameters(state, initial=False):
'''
UI: update the command-line arguments based on the interface values
'''
elements = ui.list_model_elements() # the names of the parameters elements = ui.list_model_elements() # the names of the parameters
gpu_memories = [] gpu_memories = []
@ -174,8 +176,10 @@ def update_model_parameters(state, initial=False):
shared.args.gpu_memory = None shared.args.gpu_memory = None
# UI: update the state variable with the model settings
def apply_model_settings_to_state(model, state): def apply_model_settings_to_state(model, state):
'''
UI: update the state variable with the model settings
'''
model_settings = get_model_metadata(model) model_settings = get_model_metadata(model)
if 'loader' in model_settings: if 'loader' in model_settings:
loader = model_settings.pop('loader') loader = model_settings.pop('loader')
@ -194,8 +198,10 @@ def apply_model_settings_to_state(model, state):
return state return state
# Save the settings for this model to models/config-user.yaml
def save_model_settings(model, state): def save_model_settings(model, state):
'''
Save the settings for this model to models/config-user.yaml
'''
if model == 'None': if model == 'None':
yield ("Not saving the settings because no model is loaded.") yield ("Not saving the settings because no model is loaded.")
return return