Update README.md
Browse files
README.md
CHANGED
@@ -38,6 +38,8 @@ widget:
|
|
38 |
|
39 |
- 数据集:[https://huggingface.co/datasets/LinkSoul/instruction_merge_set](https://huggingface.co/datasets/LinkSoul/instruction_merge_set)
|
40 |
|
|
|
|
|
41 |
## 快速测试
|
42 |
|
43 |
```python
|
@@ -57,12 +59,6 @@ prompt = instruction.format("用英文回答,什么是夫妻肺片?")
|
|
57 |
generate_ids = model.generate(tokenizer(prompt, return_tensors='pt').input_ids.cuda(), max_new_tokens=4096, streamer=streamer)
|
58 |
```
|
59 |
|
60 |
-
## 如何训练
|
61 |
-
|
62 |
-
```bash
|
63 |
-
python train.py --args ...
|
64 |
-
```
|
65 |
-
|
66 |
## 相关项目
|
67 |
|
68 |
- [Llama2](https://ai.meta.com/llama/)
|
|
|
38 |
|
39 |
- 数据集:[https://huggingface.co/datasets/LinkSoul/instruction_merge_set](https://huggingface.co/datasets/LinkSoul/instruction_merge_set)
|
40 |
|
41 |
+
- 训练及推理代码:[https://github.com/LinkSoul-AI/Chinese-Llama-2-7b](https://github.com/LinkSoul-AI/Chinese-Llama-2-7b)
|
42 |
+
|
43 |
## 快速测试
|
44 |
|
45 |
```python
|
|
|
59 |
generate_ids = model.generate(tokenizer(prompt, return_tensors='pt').input_ids.cuda(), max_new_tokens=4096, streamer=streamer)
|
60 |
```
|
61 |
|
|
|
|
|
|
|
|
|
|
|
|
|
62 |
## 相关项目
|
63 |
|
64 |
- [Llama2](https://ai.meta.com/llama/)
|