datasets: | |
- yuanzhoulvpi/rename_robot | |
language: | |
- zh | |
pipeline_tag: text-generation | |
1. 使用lora,给internlm模型做训练 | |
2. 训练的时候,如何让模型知道自己的身份,并且对相关问题进行拒绝回答。这里给到相关解决方案。 | |
## 模型效果 | |
这里给大家看一下,使用我这个方法训练的模型效果: | |
![](images/show_result.png) | |
![](images/show_result2.png) | |
可以看得出来: | |
1. 模型有非常明显的自我认知能力; | |
2. 模型懂得拒绝回答; | |
3. 模型对于别的问题,回答的也还可以; | |
## 训练脚本介绍 | |
GitHub训练代码:[https://github.com/yuanzhoulvpi2017/zero_nlp/tree/main/internlm-sft](https://github.com/yuanzhoulvpi2017/zero_nlp/tree/main/internlm-sft) |