fb700 commited on
Commit
c023f93
·
1 Parent(s): dffa9c0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +81 -16
README.md CHANGED
@@ -1,16 +1,36 @@
1
- license: apache-2.0
2
- ChatGLM-6B 是开源中英双语对话模型 本次训练基于ChatGLM-6B 的第一代版本,在保留了初代模型对话流畅、部署门槛较低等众多优秀特性的基础之上,本次训练使用了30万条fitness数据,40万条gpt4数据,以及30万条人类反馈数据的微调。
3
- 训练后在健康咨询,文档总结 上强于glm2,fp16运行时速度上比原模型提升20%.可以代替原有官方模型,大家可以fp16、int4、int8使用。
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  协议
5
  本仓库的代码依照 Apache-2.0 协议开源,ChatGLM2-6B 模型的权重的使用则需要遵循 Model License。
6
  授权方式,与原项目一致,未经过chatglm原开发方允许,不得用于商业用途。
7
  在原项目上的训练由智能AI用户[帛凡]于2023年基于ChatGLM独立训练的人工智能助手(严禁售卖或者商业项目,任何通过此项目产生的知识仅用于参考,作者不承担任何责任)。
 
8
  16G及以上显存用下载压缩包即lora文件使用,可支持ChatGLM原生模型和LoRA微调后的模型
9
- Usage (HuggingFace Transformers)
10
- Without textgen, you can use the model like this:
11
  First, you pass your input through the transformer model, then you get the generated sentence.
12
  Install package:
13
- pip install transformers peft
 
 
 
 
14
 
15
  import sys
16
  from peft import PeftModel
@@ -24,27 +44,66 @@ sents = ['新冠肺炎怎么预防。\n答:']
24
  for s in sents:
25
  response = model.chat(tokenizer, s, max_length=128, eos_token_id=tokenizer.eos_token_id)
26
  print(response)
27
-
28
  output:
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
 
30
  模型文件组成:
 
31
  chatglm_fitness_lora
32
  ├── adapter_config.json
33
  └── adapter_model.bin
34
-
35
  --------------------------------------------------------------------------------
36
- 16G及以下显存用户下载整个模型,可支持支持fp16、int8、int4
37
- Usage (HuggingFace Transformers)
38
- Without textgen, you can use the model like this:
39
  First, you pass your input through the transformer model, then you get the generated sentence.
40
- Install package:
41
- pip install transformers peft
 
 
 
42
 
43
  import sys
 
44
  from transformers import AutoModel, AutoTokenizer
45
  sys.path.append('..')
46
  model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True, device_map='auto')
47
- #model = model.half().quantize(4).cuda() # int4
 
48
  #model = model.half().quantize(8).cuda() # int8
49
  #model = model.half().cuda() # fp16
50
  tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
@@ -52,6 +111,12 @@ sents = ['新冠肺炎怎么预防。\n答:']
52
  for s in sents:
53
  response = model.chat(tokenizer, s, max_length=128, eos_token_id=tokenizer.eos_token_id)
54
  print(response)
55
-
56
  output:
57
-
 
 
 
 
 
 
 
1
+ tags:
2
+ chatglm
3
+ pytorch
4
+ zh
5
+ Text2Text-Generation
6
+ license: "apache-2.0"
7
+ widget:
8
+ text: "新冠肺炎怎么预防。\n答:"
9
+ --------------------------------------------------------------------------------
10
+ # ChatGLM-6B RLHF & LoRA Model
11
+
12
+ ChatGLM-6B 是开源中英双语对话模型 本次训练基于ChatGLM-6B 的第一代版本,在保留了初代模型对话流畅、部署门槛较低等众多优秀特性的基础之上开展训练。
13
+ ## 本次训练使用使用的方法
14
+
15
+ 首先,用40万条gpt4数据进行强化训练,以提高模型的基础能力;
16
+ 第二,使用30万条人类反馈数据,构建一个表达方式规范优雅的语言模式(RM模型);
17
+ 第三,在保留SFT阶段三分之一训练数据的同时,增加了30万条fitness数据,叠加RM模型,对ChatGLM-6B进行强化训练。
18
+ 最终,训练后在健康咨询,文档总结能力上不但强于glm,而且部分能力上更是强于glm2,配合“闻达”和“langchain-chatglm”等知识库项目,应用体验上对比glm和glm2均匀显著提升。
19
+ fp16运行时速度上比原模型提升20%.可以代替原有官方模型,大家可以fp16、int4、int8使用。
20
  协议
21
  本仓库的代码依照 Apache-2.0 协议开源,ChatGLM2-6B 模型的权重的使用则需要遵循 Model License。
22
  授权方式,与原项目一致,未经过chatglm原开发方允许,不得用于商业用途。
23
  在原项目上的训练由智能AI用户[帛凡]于2023年基于ChatGLM独立训练的人工智能助手(严禁售卖或者商业项目,任何通过此项目产生的知识仅用于参考,作者不承担任何责任)。
24
+ ## Usage1 16G及以上显存用下载压缩包即lora文件使用,可支持ChatGLM原生模型和LoRA微调后的模型
25
  16G及以上显存用下载压缩包即lora文件使用,可支持ChatGLM原生模型和LoRA微调后的模型
26
+ (HuggingFace Transformers)
 
27
  First, you pass your input through the transformer model, then you get the generated sentence.
28
  Install package:
29
+ ```
30
+ pip install transformers
31
+ ```
32
+
33
+ ```python
34
 
35
  import sys
36
  from peft import PeftModel
 
44
  for s in sents:
45
  response = model.chat(tokenizer, s, max_length=128, eos_token_id=tokenizer.eos_token_id)
46
  print(response)
47
+ ```
48
  output:
49
+ ```shell
50
+ 例一:常识综合问答
51
+ 问: 新冠肺炎怎么预防
52
+ 答:新冠肺炎是一种由新型冠状病毒引起的疾病,可以通过飞沫传播和接触传播。以下是一些预防新冠肺炎的建议:
53
+ 戴口罩:佩戴口罩可以防止呼吸道飞沫传播,减少感染风险。在公共场合、公共交通工具上、与他人交往时,应佩戴医用口罩或N95口罩。
54
+ 勤洗手:经常用肥皂和温水洗手,尤其在接触公共场所、物品或与他人交往后,或咳嗽、打喷嚏、擤鼻涕后等。
55
+ 保持社交距离:尽量避免近距离接触他人,保持至少1米的社交距离。
56
+ 避免聚集:尽量避免参加人员密集的聚会、活动或旅行。
57
+ 加强室内通风:在室内保持良好的通风可以降低病毒在空气中的浓度。
58
+ 健康生活方式:保持充足的睡眠、均衡的饮食、适量的运动等健康生活方式可以增强身体免疫力,预防感染。
59
+ 接种疫苗:疫苗接种是预防新冠肺炎最有效的方法之一。如果有机会,应接种疫苗。
60
+ 预防新冠肺炎需要个人和社会的共同努力,遵守公共卫生规定,加强个人防护,减少接触风险。
61
+ 例二:较为专业的医疗和心理学问答
62
+ 问:系统性红斑狼疮的危害和治疗方法是什么?
63
+ 答:系统性红斑狼疮(Systemic Lupus Erythematosus,SLE)是一种自身免疫性疾病,可以影响身体的多个器官和组织。SLE的危害包括:
64
+ 1. 心血管疾病:SLE可以引起高血压、心脏病和中风等心血管疾病。
65
+ 2. 肾脏疾病:SLE可以引起肾小球肾炎和肾衰竭等肾脏疾病。
66
+ 3. 口腔溃疡:SLE可以引起口腔和舌头溃疡等口腔疾病。
67
+ 4. 皮肤疾病:SLE可以引起皮疹、红斑狼疮面容和光敏感等皮肤疾病。
68
+
69
+ 治疗方法包括:
70
+ 1. 药物治疗:SLE可以使用非甾体抗炎药、激素类药物和免疫抑制剂等药物治疗。
71
+ 2. 支持性治疗:SLE可以使用营养支持、心理治疗和支持性护理等支持性治疗。
72
+ 3. 手术治疗:在一些情况下,SLE可能需要手术治疗,如肾脏移植等。
73
+
74
+ SLE是一种严重的疾病,需要患者及时接受专业治疗。患者需要积极配合医生的治疗,保持良好的生活习惯和饮食习惯,定期进行检查和监测。
75
+ 例三:无限多轮次的智能对话(第一轮见例二)
76
+ 问:早期激素和免疫抑制剂用药建议是什么?
77
+ 答:系统性红斑狼疮的早期治疗非常重要,因为病情的严重程度和进展速度可能会影响患者的生活质量和生命安全。对于系统性红斑狼疮的早期治疗,激素和免疫抑制剂的使用是非常重要的。以下是一些早期激素和免疫抑制剂用药的建议:
78
+ 1. 激素类药物:激素类药物是系统性红斑狼疮的主要治疗方法之一。在使用激素类药物时,需要遵循医生的建议,根据病情和身体状况进行剂量和用药时间的调整。一般来说,激素类药物需要长期使用,以避免药物副作用和病情的复发。
79
+ 2. 免疫抑制剂:免疫抑制剂是系统性红斑狼疮另一种重要的治疗方法。在使用免疫抑制剂时,需要遵循医生的建议,根据病情和身体状况进行用药时间和剂量的调整。免疫抑制剂可以用于治疗系统性红斑狼疮的各种症状和并发症,如肾脏疾病、口腔溃疡、皮肤疾病等。
80
+ 3. 其他治疗:除了激素和免疫抑制剂之外,系统性红斑狼疮的治疗还包括其他治疗方法,如支持性治疗、手术治疗等。在使用这些治疗方法时,需要遵循医生的建议,根据病情和身体状况进行用药和治疗。
81
+
82
+ 早期激素和免疫抑制剂的使用非常重要,需要患者积极配合医生的治疗,遵循医生的建议,定期进行检查和监测,以确保病情得到有效控制和生活质量得到保障。
83
+ ```
84
 
85
  模型文件组成:
86
+ ```
87
  chatglm_fitness_lora
88
  ├── adapter_config.json
89
  └── adapter_model.bin
90
+ ```
91
  --------------------------------------------------------------------------------
92
+ ## Usage2 16G及以下显存用户下载整个模型,可支持支持fp16、int8、int4
 
 
93
  First, you pass your input through the transformer model, then you get the generated sentence.
94
+ ```
95
+ pip install transformers
96
+ ```
97
+
98
+ ```python
99
 
100
  import sys
101
+ from peft import PeftModel
102
  from transformers import AutoModel, AutoTokenizer
103
  sys.path.append('..')
104
  model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True, device_map='auto')
105
+ model = PeftModel.from_pretrained(model, "model/chatglm_fitness_lora") # lora文件保存目录
106
+ model = model.half().quantize(4).cuda() # int4
107
  #model = model.half().quantize(8).cuda() # int8
108
  #model = model.half().cuda() # fp16
109
  tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
 
111
  for s in sents:
112
  response = model.chat(tokenizer, s, max_length=128, eos_token_id=tokenizer.eos_token_id)
113
  print(response)
114
+ ```
115
  output:
116
+ ```shell
117
+ 例四:超越chatglm2 的总结归纳能力
118
+ 问:请用简短的语言总结下面的文字:
119
+ 大语言模型是指能够生成、理解和处理自然语言的高度智能化的计算机模型。这些模型使用深度学习技术,尤其是循环神经网络(RNN)或变种,如长短期记忆(LSTM)或注意力机制(attention mechanism),从大规模文本语料库中进行训练。
120
+ 大语言模型的训练过程通常基于预测下一个单词或字符的任务。通过对大量文本数据进行训练,模型能够学习到语言的潜在模式、结构和语义含义。这使得大语言模型能够产生流畅、连贯的文本,回答问题,完成翻译任务,生成代码等。
121
+ 答:大语言模型是一种使用深度学习技术训练的计算机模型,能够生成、理解和处理自然语言。通过训练大量文本数据,大语言模型能够产生流畅、连贯的文本,回答问题,完成翻译任务,生成代码等。
122
+ ```