shiyemin2 commited on
Commit
1017bec
1 Parent(s): 0dd2971

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -6
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/)