Added default Qwen modelpath

This commit is contained in:
Angel-HF
2024-04-06 12:53:08 +02:00
committed by Pelochus
parent 2f64326612
commit 6cb14da924

View File

@@ -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)