hiyouga/Yi-Agent-6B

A bilingual model with function calling abilities.

• Fine-tuned from: https://huggingface.co/01-ai/Yi-6B
• Instruction-following datasets used: alpaca, open assistant, glaive function calling
• Training framework: LLaMA-Factory

Usage:

Launch a RESTFul API by using the script in LLaMA-Factory

python src/api_demo.py --template default --model_name_or_path hiyouga/Yi-Agent-6B

Test with:

https://github.com/hiyouga/LLaMA-Factory/blob/main/tests/test_toolcall.py

You can reproduce our results with the step-by-step (Chinese) guide:

https://zhuanlan.zhihu.com/p/678989191