镜像自地址
https://github.com/SCIR-HI/Med-ChatGLM.git
已同步 2025-12-05 22:26:50 +00:00
Update modeling_chatglm.py
Fix the ValueError about the ``150000/150001''
这个提交包含在:
@@ -828,7 +828,7 @@ class ChatGLMModel(ChatGLMPreTrainedModel):
|
||||
if past_key_values is None:
|
||||
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
|
||||
use_gmask = False if MASK in input_ids else gMASK
|
||||
seq = input_ids[0].tolist()
|
||||
@@ -972,7 +972,7 @@ class ChatGLMForConditionalGeneration(ChatGLMPreTrainedModel):
|
||||
**kwargs
|
||||
) -> dict:
|
||||
|
||||
MASK, gMASK = 150000, 150001
|
||||
MASK, gMASK = 130000, 130001
|
||||
mask_token = MASK if MASK in input_ids else gMASK
|
||||
use_gmask = False if MASK in input_ids else gMASK
|
||||
seq = input_ids[0].tolist()
|
||||
|
||||
在新工单中引用
屏蔽一个用户