wassname/Clover-Edition
1
0
Fork 0
mirror of https://github.com/wassname/Clover-Edition.git synced 2026-09-12 12:04:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
nickwalton
2019-10-30 19:00:21 -06:00
parent acafa6d292
commit 74a4a05504
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/ctrl/ctrl_generator.py
+1 -3
View File
@@ -135,7 +135,7 @@ class CTRLGenerator():
self.temperature=temperature
self.nucleusprob = nucleus_prob
self.penalty = 1.2
self.penalty = 1.1
self.topk=topk
def configure_verb_probs(self, probabilities, options):
@@ -212,8 +212,6 @@ class CTRLGenerator():
penalized_so_far = set()
for _ in range(token + 1):
generated_token = tokens_generated[0][_]
if generated_token in penalized_so_far:
continue
penalized_so_far.add(generated_token)
prompt_logits[_token][generated_token] /= self.penalty