From cc6685e93c376516d4e8ec9dc880ce735eb196c4 Mon Sep 17 00:00:00 2001 From: numb3r3 Date: Thu, 20 Jul 2023 11:54:06 +0800 Subject: [PATCH] hotfix: revert use_fast --- open_gpt/models/loading.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/open_gpt/models/loading.py b/open_gpt/models/loading.py index e5653f3..1c2a9ab 100644 --- a/open_gpt/models/loading.py +++ b/open_gpt/models/loading.py @@ -24,7 +24,7 @@ def load_model_and_tokenizer( tokenizer = AutoTokenizer.from_pretrained( tokenizer_name_or_path or model_name_or_path, trust_remote_code=True, - use_fast=use_fast, + # use_fast=use_fast, ) if tokenizer.pad_token is None: