From 08d880141d7450099a579e00877e3318be26da5f Mon Sep 17 00:00:00 2001 From: duzx16 Date: Fri, 31 Mar 2023 16:32:34 +0800 Subject: [PATCH] Fix revision for loading model --- ptuning/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ptuning/main.py b/ptuning/main.py index d82fccc..5b23234 100644 --- a/ptuning/main.py +++ b/ptuning/main.py @@ -110,7 +110,7 @@ def main(): tokenizer = AutoTokenizer.from_pretrained(model_args.model_name_or_path, trust_remote_code=True) - model = AutoModel.from_pretrained(model_args.model_name_or_path, config=config, revision=True, trust_remote_code=True) + model = AutoModel.from_pretrained(model_args.model_name_or_path, config=config, trust_remote_code=True) model = model.half() if model_args.quantization_bit is not None: