From 6cb14da9242b5ac5d6016ba2de99203d034a7ebb Mon Sep 17 00:00:00 2001 From: Angel-HF <49835792+Pelochus@users.noreply.github.com> Date: Sat, 6 Apr 2024 12:53:08 +0200 Subject: [PATCH] Added default Qwen modelpath --- rkllm-toolkit/examples/huggingface/test.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/rkllm-toolkit/examples/huggingface/test.py b/rkllm-toolkit/examples/huggingface/test.py index c253fe4..11519fd 100644 --- a/rkllm-toolkit/examples/huggingface/test.py +++ b/rkllm-toolkit/examples/huggingface/test.py @@ -5,7 +5,8 @@ https://huggingface.co/Qwen/Qwen-1_8B-Chat Download the Qwen model from the above website. ''' -modelpath = '/path/to/your/model' +# Default option. Works only if in the same directory +modelpath = 'Qwen-1_8B-Chat' llm = RKLLM() # Load model @@ -25,5 +26,3 @@ ret = llm.export_rkllm("./qwen.rkllm") if ret != 0: print('Export model failed!') exit(ret) - -