From b69b1ce308e879eb99c5a58945180d8157c9941a Mon Sep 17 00:00:00 2001 From: Nick Date: Thu, 31 Oct 2019 03:05:27 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index d325d5e..4be6f57 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -265,9 +265,6 @@ class CTRLGenerator(): tf.random.categorical(np.expand_dims(prompt_logits[_token][pruned_list], 0), num_samples=1).numpy()) idx = pruned_list[chosen_idx] - # assign the token for generation - tokens_generated[0][token + 1] = idx - return idx def generate(self, prompt, options=None): @@ -307,6 +304,8 @@ class CTRLGenerator(): idx = self.generate_next_token(token, tokens_generated, options, num_new_lines, token_num, first_token=first_token, forbid_newline=True) + # assign the token for generation + tokens_generated[0][token + 1] = idx if debug_print: print(repr(self.idx2word[idx]), end="_")