internlm
/

internlm2-chat-20b-sft

Text Generation

Model card Files Files and versions Community

x54-729 commited on Jan 24

Commit

e175756

•

1 Parent(s): 52f6922

small

Files changed (1) hide show

modeling_internlm2.py +1 -1

modeling_internlm2.py CHANGED Viewed

@@ -1215,7 +1215,7 @@ class InternLM2ForCausalLM(InternLM2PreTrainedModel):
                 self.query = query
                 self.history = history
                 self.response = ""
-                self.chat = []
                 self.received_inputs = False
                 self.queue.put((self.response, history + [(self.query, self.response)]))

                 self.query = query
                 self.history = history
                 self.response = ""
+                self.cache = []
                 self.received_inputs = False
                 self.queue.put((self.response, history + [(self.query, self.response)]))