Minor bug fix

2023-04-09 23:45:41 -03:00 · 2023-04-09 23:45:41 -03:00 · 1911504f82
commit 1911504f82
parent 8178fde2cb
1 changed files with 1 additions and 1 deletions
--- a/modules/models.py
+++ b/modules/models.py
@ -186,7 +186,7 @@ def load_model(model_name):
            tokenizer.bos_token_id = 1
            tokenizer.pad_token_id = 0
        except:
-            continue
+            pass
    else:
        tokenizer = AutoTokenizer.from_pretrained(Path(f"{shared.args.model_dir}/{shared.model_name}/"))
    tokenizer.truncation_side = 'left'