From 1911504f8205bfae5203185d89d7860a0f930974 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Sun, 9 Apr 2023 23:45:41 -0300 Subject: [PATCH] Minor bug fix --- modules/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/models.py b/modules/models.py index 32c9c348..7ec93df8 100644 --- 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'