mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-10-30 12:50:18 +08:00
added telemetry
This commit is contained in:
1 parent
8a200b62b2
commit
2ce06037f8
3 files changed
+3
-3
No files matched your search
@@ -10,7 +10,7 @@ import numpy as np
|
||||
|
||||
class GPT2Generator:
|
||||
|
||||
def __init__(self, generate_num=60, temperature=0.3, top_k=40, top_p=0.9):
|
||||
def __init__(self, generate_num=60, temperature=0.4, top_k=40, top_p=0.9):
|
||||
self.generate_num=generate_num
|
||||
self.temp = temperature
|
||||
self.top_k = top_k
|
||||
|
||||
@@ -17,7 +17,7 @@ def penalize_used(logits, output):
|
||||
|
||||
return tf.compat.v1.where(
|
||||
bool_tensor,
|
||||
logits / 1.2,
|
||||
logits * 0.85,
|
||||
logits)
|
||||
|
||||
|
||||
|
||||
Reference in new issue
Block a user