|
--- |
|
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. |
|
|
|
- Fine-tuned from: https://huggingface.co/01-ai/Yi-6B |
|
- Instruction-following datasets used: alpaca, open assistant, glaive function calling |
|
- Training framework: [LLaMA-Factory](https://github.com/hiyouga/LLaMA-Factory) |
|
|
|
Usage: |
|
|
|
Launch a RESTFul API by using the script in [LLaMA-Factory](https://github.com/hiyouga/LLaMA-Factory) |
|
|
|
```bash |
|
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 |
|
|
