starlight-apk/Qwen
1
0
Fork 0
mirror of https://github.com/QwenLM/Qwen.git synced 2026-05-20 16:35:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

format line

Browse Source
This commit is contained in:
yangapku
2023-08-24 14:14:30 +08:00
parent 1b6758e157
commit 5c49d35a7e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
openai_api.py
View File

@@ -202,7 +202,7 @@ def _get_args():
if __name__ == "__main__": if __name__ == "__main__":
args = _get_args() args = _get_args()
tokenizer = AutoTokenizer.from_pretrained( tokenizer = AutoTokenizer.from_pretrained(
args.checkpoint_path, trust_remote_code=True, resume_download=True, args.checkpoint_path, trust_remote_code=True, resume_download=True,
) )
@@ -218,7 +218,7 @@ if __name__ == "__main__":
trust_remote_code=True, trust_remote_code=True,
resume_download=True, resume_download=True,
).eval() ).eval()
model.generation_config = GenerationConfig.from_pretrained( model.generation_config = GenerationConfig.from_pretrained(
args.checkpoint_path, trust_remote_code=True, resume_download=True, args.checkpoint_path, trust_remote_code=True, resume_download=True,
) )