int/Med-ChatGLM
1
0
派生 0
镜像自地址 https://github.com/SCIR-HI/Med-ChatGLM.git 已同步 2025-12-06 14:46:49 +00:00
代码 工单 软件包 项目 版本发布 百科 动态

Update modeling_chatglm.py

浏览代码 
Fix the ValueError about the ``150000/150001''
这个提交包含在：
Haochun Wang
2023-04-27 14:29:59 +08:00
提交者 GitHub
父节点 cb9d827380
当前提交 61471075ed
共有 1 个文件被更改，包括 2 次插入 和 2 次删除
下载 Patch 文件 下载 Diff 文件 展开所有文件 折叠所有文件

4
modeling_chatglm.py
查看文件

@@ -828,7 +828,7 @@ class ChatGLMModel(ChatGLMPreTrainedModel):
if past_key_values is None: if past_key_values is None:
past_key_values = tuple([None] * len(self.layers)) past_key_values = tuple([None] * len(self.layers))
MASK, gMASK = 150000, 150001 MASK, gMASK = 130000, 130001
mask_token = MASK if MASK in input_ids else gMASK mask_token = MASK if MASK in input_ids else gMASK
use_gmask = False if MASK in input_ids else gMASK use_gmask = False if MASK in input_ids else gMASK
seq = input_ids[0].tolist() seq = input_ids[0].tolist()
@@ -972,7 +972,7 @@ class ChatGLMForConditionalGeneration(ChatGLMPreTrainedModel):
**kwargs **kwargs
) -> dict: ) -> dict:
MASK, gMASK = 150000, 150001 MASK, gMASK = 130000, 130001
mask_token = MASK if MASK in input_ids else gMASK mask_token = MASK if MASK in input_ids else gMASK
use_gmask = False if MASK in input_ids else gMASK use_gmask = False if MASK in input_ids else gMASK
seq = input_ids[0].tolist() seq = input_ids[0].tolist()