Explorar o código

Minor bug fix

oobabooga %!s(int64=2) %!d(string=hai) anos
pai
achega
1911504f82
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      modules/models.py

+ 1 - 1
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'