update

2026-09-10 11:40:48 +08:00 · 2019-10-30 10:46:18 -06:00
parent ea2e2af206
commit d5ab21b025
1 changed files with 1 additions and 1 deletions
@@ -291,7 +291,7 @@ class CTRLGenerator():
        num_new_lines = 0
        for token in range(len(text) - 1, total_text_len - 1):
            idx = self.generate_next_token(token, tokens_generated, options, num_new_lines, token_num, first_token=first_token)
-            if self.idx2word[idx] is "\n":
+            if self.idx2word[idx] == '\n':
                return self.result_replace(result)

            print(repr(self.idx2word[idx]), end="_")