From 5c49d35a7e33a531075168729bade716389ad6a8 Mon Sep 17 00:00:00 2001 From: yangapku Date: Thu, 24 Aug 2023 14:14:30 +0800 Subject: [PATCH] format line --- openai_api.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/openai_api.py b/openai_api.py index 52da00b..42b0841 100644 --- a/openai_api.py +++ b/openai_api.py @@ -202,7 +202,7 @@ def _get_args(): if __name__ == "__main__": args = _get_args() - + tokenizer = AutoTokenizer.from_pretrained( args.checkpoint_path, trust_remote_code=True, resume_download=True, ) @@ -218,7 +218,7 @@ if __name__ == "__main__": trust_remote_code=True, resume_download=True, ).eval() - + model.generation_config = GenerationConfig.from_pretrained( args.checkpoint_path, trust_remote_code=True, resume_download=True, )