github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update modeling_internlm.py 

Correct spelling mistakes: chche -> cache
pull/560/head
djsaber 2023-12-28 13:36:12 +08:00 committed by GitHub
parent 508711cc97
commit e7686e7fb8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tools/transformers/modeling_internlm.py
View File

@ -844,7 +844,7 @@ class InternLMForCausalLM(InternLMPreTrainedModel):
self.query = query self.query = query
self.history = history self.history = history
self.response = "" self.response = ""
self.chche = [] self.cache = []
self.received_inputs = False self.received_inputs = False
self.queue.put((self.response, history + [(self.query, self.response)])) self.queue.put((self.response, history + [(self.query, self.response)]))
@ -859,8 +859,8 @@ class InternLMForCausalLM(InternLMPreTrainedModel):
self.received_inputs = True self.received_inputs = True
return return
self.chche.extend(value.tolist()) self.cache.extend(value.tolist())
token = self.tokenizer.decode(self.chche, skip_special_tokens=True) token = self.tokenizer.decode(self.cache, skip_special_tokens=True)
if "<EFBFBD>" in token and len(token) <= 5: if "<EFBFBD>" in token and len(token) <= 5:
return return
@ -868,7 +868,7 @@ class InternLMForCausalLM(InternLMPreTrainedModel):
self.response = self.response + token self.response = self.response + token
history = self.history + [(self.query, self.response)] history = self.history + [(self.query, self.response)]
self.queue.put((self.response, history)) self.queue.put((self.response, history))
self.chche = [] self.cache = []
else: else:
self.end() self.end()