Update README.md
Browse files
README.md
CHANGED
@@ -2,4 +2,35 @@
|
|
2 |
license: other
|
3 |
license_name: yi-license
|
4 |
license_link: https://huggingface.co/01-ai/Yi-6B/blob/main/LICENSE
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
license: other
|
3 |
license_name: yi-license
|
4 |
license_link: https://huggingface.co/01-ai/Yi-6B/blob/main/LICENSE
|
5 |
+
datasets:
|
6 |
+
- tatsu-lab/alpaca
|
7 |
+
- OpenAssistant/oasst1
|
8 |
+
- glaiveai/glaive-function-calling-v2
|
9 |
+
language:
|
10 |
+
- en
|
11 |
+
- zh
|
12 |
+
library_name: transformers
|
13 |
+
tags:
|
14 |
+
- lora
|
15 |
+
- llama-factory
|
16 |
+
pipeline_tag: text-generation
|
17 |
+
inference: false
|
18 |
---
|
19 |
+
|
20 |
+
A bilingual model with function calling abilities.
|
21 |
+
|
22 |
+
- Fine-tuned from: https://huggingface.co/01-ai/Yi-6B
|
23 |
+
- Instruction-following datasets used: alpaca, open assistant, glaive function calling
|
24 |
+
- Training framework: [LLaMA-Factory](https://github.com/hiyouga/LLaMA-Factory)
|
25 |
+
|
26 |
+
Usage:
|
27 |
+
|
28 |
+
Launch a RESTFul API by using the script in [LLaMA-Factory](https://github.com/hiyouga/LLaMA-Factory)
|
29 |
+
|
30 |
+
```bash
|
31 |
+
python src/api_demo.py --template default --model_name_or_path hiyouga/Yi-Agent-6B
|
32 |
+
```
|
33 |
+
|
34 |
+
Test tool using with:
|
35 |
+
|
36 |
+
https://github.com/hiyouga/LLaMA-Factory/blob/main/tests/test_toolcall.py
|