mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-09-11 11:51:53 +08:00
update
This commit is contained in:
@@ -18,7 +18,7 @@ tf.logging.set_verbosity(tf.logging.ERROR)
|
||||
|
||||
class SimpleGenerator:
|
||||
|
||||
def __init__(self, generate_num=40, temperature=0.9, top_k=40, top_p=0.0):
|
||||
def __init__(self, generate_num=65, temperature=0.9, top_k=40, top_p=0.0):
|
||||
self.generate_num=generate_num
|
||||
self.temp = temperature
|
||||
self.top_k = top_k
|
||||
|
||||
@@ -41,7 +41,13 @@ class Story():
|
||||
self.results.append(story_block)
|
||||
|
||||
def latest_result(self):
|
||||
if len(self.results) >= 2:
|
||||
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:
|
||||
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]
|
||||
|
||||
Reference in New Issue
Block a user