Text Generation
Transformers
Safetensors
Chinese
English
qwen
conversational
custom_code
yuyijiong commited on
Commit
ded233f
1 Parent(s): f649fd0

Update modeling_qwen_yarn.py

Browse files
Files changed (1) hide show
  1. modeling_qwen_yarn.py +0 -2
modeling_qwen_yarn.py CHANGED
@@ -1156,8 +1156,6 @@ class QWenLMHeadModel(QWenPreTrainedModel):
1156
  output = (lm_logits,) + transformer_outputs[1:]
1157
  return ((loss,) + output) if loss is not None else output
1158
 
1159
- if self.training:
1160
- lm_logits=None
1161
  return CausalLMOutputWithPast(
1162
  loss=loss,
1163
  logits=lm_logits,
 
1156
  output = (lm_logits,) + transformer_outputs[1:]
1157
  return ((loss,) + output) if loss is not None else output
1158
 
 
 
1159
  return CausalLMOutputWithPast(
1160
  loss=loss,
1161
  logits=lm_logits,