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

update

Browse Source
This commit is contained in:
Nick committed 2019-09-19 22:44:08 -06:00
1 parent 35eaefaa32
commit da015e2149
3 files changed
+9 -1
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

generator/ctrl/ctrl_generator.py
+2
View File
@@ -256,6 +256,8 @@ class CTRLGenerator():
# os.system("clear")
tokens_generated_so_far = ' '.join([self.idx2word[c] for c in tokens_generated[0].squeeze()[:token + 2]])
tokens_generated_so_far = re.sub('(@@ )', '', string=tokens_generated_so_far)
tokens_generated_so_far = re.sub('(@@ ?$)', '', string=tokens_generated_so_far)
result = tokens_generated_so_far[prompt_length:]