Maxtimer97 commited on
Commit
bf53a99
·
verified ·
1 Parent(s): bb18c9e

Removed GenerationMixing

Browse files
Files changed (1) hide show
  1. modeling_chatglm.py +1 -1
modeling_chatglm.py CHANGED
@@ -1153,7 +1153,7 @@ class ChatGLMModel(ChatGLMPreTrainedModel):
1153
  )
1154
 
1155
 
1156
- class ChatGLMForConditionalGeneration(ChatGLMPreTrainedModel, GenerationMixin):
1157
  def __init__(self, config: ChatGLMConfig, empty_init=True, device=None):
1158
  super().__init__(config)
1159
 
 
1153
  )
1154
 
1155
 
1156
+ class ChatGLMForConditionalGeneration(ChatGLMPreTrainedModel): #, GenerationMixin):
1157
  def __init__(self, config: ChatGLMConfig, empty_init=True, device=None):
1158
  super().__init__(config)
1159