Cleanup: set shared.model_name only once

2023-12-08 06:35:23 -08:00 · 2023-12-08 06:35:23 -08:00 · 2a335b8aa7
commit 2a335b8aa7
parent 62d59a516f
4 changed files with 3 additions and 5 deletions
--- a/modules/models.py
+++ b/modules/models.py
@ -58,6 +58,7 @@ def load_model(model_name, loader=None):
    t0 = time.time()

    shared.is_seq2seq = False
+    shared.model_name = model_name
    load_func_map = {
        'Transformers': huggingface_loader,
        'AutoGPTQ': AutoGPTQ_loader,