From 169d42d3f82d4cff113b272593c81fdd218f30e9 Mon Sep 17 00:00:00 2001 From: Nick Date: Tue, 24 Sep 2019 14:30:27 -0600 Subject: [PATCH] fixed mental issue? --- generator/ctrl/ctrl_generator.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 9d40914..db2a759 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -212,10 +212,7 @@ class CTRLGenerator(): num_new_lines = 0 for token in range(len(text) - 1, total_text_len - 1): - if first_token: - temperature = 0.9 - else: - temperature = self.temperature + temperature = self.temperature # get the logits from the prediction function # the logic here is a bit convoluted because we are allowing generation past 512 tokens @@ -329,4 +326,6 @@ class CTRLGenerator(): result = tokens_generated_so_far[prompt_length:] first_token = False + print("tokens generated so far is ") + print(tokens_generated_so_far) return self.result_replace(result)