Yi-Agent-6B / README.md
hiyouga's picture
Update README.md
ecb6495 verified
metadata
license: other
license_name: yi-license
license_link: https://huggingface.co/01-ai/Yi-6B/blob/main/LICENSE
datasets:
  - tatsu-lab/alpaca
  - OpenAssistant/oasst1
  - glaiveai/glaive-function-calling-v2
language:
  - en
  - zh
library_name: transformers
tags:
  - lora
  - llama-factory
pipeline_tag: text-generation
inference: false

A bilingual model with function calling abilities.

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