mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-10-30 12:50:18 +08:00
update
This commit is contained in:
1 parent
207b3b7b11
commit
bdd092e641
2 files changed
+2
-2
No files matched your search
@@ -11,7 +11,7 @@ tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
|
||||
|
||||
class GPT2Generator:
|
||||
|
||||
def __init__(self, generate_num=80, temperature=0.3, top_k=40, top_p=0.8):
|
||||
def __init__(self, generate_num=64, temperature=0.3, top_k=40, top_p=0.8):
|
||||
self.generate_num=generate_num
|
||||
self.temp = temperature
|
||||
self.top_k = top_k
|
||||
|
||||
Reference in new issue
Block a user