Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-25 04:30:48 +01:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
xingjun.wang
2023-10-22 19:06:24 +08:00
parent c075d13aa6
commit bea2ce7a82
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modelscope/models/nlp/chatglm2/text_generation.py
View File

@@ -1115,9 +1115,9 @@ class ChatGLM2ForConditionalGeneration(ChatGLMPreTrainedModel):
output = (lm_logits, ) + transformer_outputs[1:]
return ((loss, ) + output) if loss is not None else output
print(f'\n>>lm_logits in ChatGLM2ForConditionalGeneration forward:'
f'\n >data:\n{lm_logits} '
f'\n >shape:\n{lm_logits.shape}\n')
# print(f'\n>>lm_logits in ChatGLM2ForConditionalGeneration forward:'
# f'\n >data:\n{lm_logits} '
# f'\n >shape:\n{lm_logits.shape}\n')
return CausalLMOutputWithPast(
loss=loss,