Update README.md
Browse files
README.md
CHANGED
@@ -15,7 +15,7 @@ license: apache-2.0
|
|
15 |
|
16 |
|
17 |
## 最近更新
|
18 |
-
- 👏🏻 2023.06.06: 扁鹊-2.0模型开源,详情见[BianQue-2.0](https://huggingface.co/scutcyr/BianQue-2
|
19 |
- 👏🏻 2023.06.06: 具备共情与倾听能力的灵心健康大模型SoulChat发布,详情见:[灵心健康大模型SoulChat:通过长文本咨询指令与多轮共情对话数据集的混合微调,提升大模型的“共情”能力 ](https://huggingface.co/scutcyr/SoulChat)。
|
20 |
- 👏🏻 2023.04.22: 基于扁鹊-1.0模型的医疗问答系统Demo,详情访问:[https://huggingface.co/spaces/scutcyr/BianQue](https://huggingface.co/spaces/scutcyr/BianQue)
|
21 |
- 👏🏻 2023.04.22: 扁鹊-1.0版本模型发布,详情见:[扁鹊-1.0:通过混合指令和多轮医生问询数据集的微调,提高医疗聊天模型的“问”能力(BianQue-1.0: Improving the "Question" Ability of Medical Chat Model through finetuning with Hybrid Instructions and Multi-turn Doctor QA Datasets)](https://huggingface.co/scutcyr/BianQue-1.0)
|
@@ -59,7 +59,7 @@ from transformers import AutoModel, AutoTokenizer
|
|
59 |
# GPU设置
|
60 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
61 |
# 加载模型与tokenizer
|
62 |
-
model_name_or_path = 'scutcyr/BianQue-2
|
63 |
model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True).half()
|
64 |
model.to(device)
|
65 |
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True)
|
@@ -94,7 +94,7 @@ os.environ['CUDA_VISIBLE_DEVICES'] = '1'
|
|
94 |
```
|
95 |
可以通过更改以下代码指定模型路径为本地路径:
|
96 |
```python
|
97 |
-
model_name_or_path = "scutcyr/BianQue-2
|
98 |
```
|
99 |
|
100 |
我们还提供了[bianque_v1_app.py](https://github.com/scutcyr/BianQue/blob/main/bianque_v1_app.py)作为BianQue-1.0模型的使用示例,以及[bianque_v1_v2_app.py](./bianque_v1_v2_app.py)作为BianQue-1.0模型和BianQue-2.0模型的联合使用示例。
|
|
|
15 |
|
16 |
|
17 |
## 最近更新
|
18 |
+
- 👏🏻 2023.06.06: 扁鹊-2.0模型开源,详情见[BianQue-2.0](https://huggingface.co/scutcyr/BianQue-2)。
|
19 |
- 👏🏻 2023.06.06: 具备共情与倾听能力的灵心健康大模型SoulChat发布,详情见:[灵心健康大模型SoulChat:通过长文本咨询指令与多轮共情对话数据集的混合微调,提升大模型的“共情”能力 ](https://huggingface.co/scutcyr/SoulChat)。
|
20 |
- 👏🏻 2023.04.22: 基于扁鹊-1.0模型的医疗问答系统Demo,详情访问:[https://huggingface.co/spaces/scutcyr/BianQue](https://huggingface.co/spaces/scutcyr/BianQue)
|
21 |
- 👏🏻 2023.04.22: 扁鹊-1.0版本模型发布,详情见:[扁鹊-1.0:通过混合指令和多轮医生问询数据集的微调,提高医疗聊天模型的“问”能力(BianQue-1.0: Improving the "Question" Ability of Medical Chat Model through finetuning with Hybrid Instructions and Multi-turn Doctor QA Datasets)](https://huggingface.co/scutcyr/BianQue-1.0)
|
|
|
59 |
# GPU设置
|
60 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
61 |
# 加载模型与tokenizer
|
62 |
+
model_name_or_path = 'scutcyr/BianQue-2'
|
63 |
model = AutoModel.from_pretrained(model_name_or_path, trust_remote_code=True).half()
|
64 |
model.to(device)
|
65 |
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True)
|
|
|
94 |
```
|
95 |
可以通过更改以下代码指定模型路径为本地路径:
|
96 |
```python
|
97 |
+
model_name_or_path = "scutcyr/BianQue-2"
|
98 |
```
|
99 |
|
100 |
我们还提供了[bianque_v1_app.py](https://github.com/scutcyr/BianQue/blob/main/bianque_v1_app.py)作为BianQue-1.0模型的使用示例,以及[bianque_v1_v2_app.py](./bianque_v1_v2_app.py)作为BianQue-1.0模型和BianQue-2.0模型的联合使用示例。
|