Cleanup: set shared.model_name only once

This commit is contained in:
oobabooga 2023-12-08 06:35:23 -08:00
parent 62d59a516f
commit 2a335b8aa7
4 changed files with 3 additions and 5 deletions

View file

@ -55,7 +55,6 @@ def _load_model(data):
setattr(shared.args, k, args[k])
shared.model, shared.tokenizer = load_model(model_name)
shared.model_name = model_name
# Update shared.settings with custom generation defaults
if settings: