This commit is contained in:
root
2019-11-16 04:23:56 +00:00
parent 9934ab4251
commit aa39f5d2ca
2 changed files with 6 additions and 8 deletions
+1 -7
View File
@@ -41,13 +41,7 @@ class Story():
self.results.append(story_block)
def latest_result(self):
if len(self.results) >= 4:
return self.context + self.actions[-4] + self.results[-4] + self.actions[-3] + self.results[-3] + \
self.actions[-2] + self.results[-2] + self.actions[-1] + self.results[-1]
elif len(self.results) >= 3:
return self.context + self.actions[-3] + self.results[-3] + self.actions[-2] + self.results[-2] + \
self.actions[-1] + self.results[-1]
elif len(self.results) >= 2:
if len(self.results) >= 2:
return self.context + self.actions[-2] + self.results[-2] + self.actions[-1] + self.results[-1]
elif len(self.results) >= 1:
return self.context + self.actions[-1] + self.results[-1]
+5 -1
View File
@@ -80,7 +80,11 @@ def cut_trailing_sentence(text):
et_token = text.rfind("<|endoftext|>")
if et_token != -1:
last_punc = min(last_punc, et_token)
last_punc = min(last_punc, et_token+1)
act_token = text.find(">")
if act_token != -1:
last_punc = min(last_punc, act_token+1)
if last_punc > 0:
text = text[0:last_punc+1]