This commit is contained in:
Nick
2019-09-20 16:16:31 -06:00
parent d134503c94
commit 84667ce811
+4 -4
View File
@@ -147,9 +147,9 @@ class CTRLGenerator():
prompt = self.control_code + prompt
print("******************************")
print(" DEBUG:: Prompt to generate by is \n", prompt)
print("******************************")
# print("******************************")
# print(" DEBUG:: Prompt to generate by is \n", prompt)
# print("******************************")
prompt_length = len(prompt)
@@ -257,7 +257,7 @@ class CTRLGenerator():
# like which one was chosen,
# what the top25 were,
# here is your opportunity.
print('chosen:', repr(self.idx2word[idx]))
#print('chosen:', repr(self.idx2word[idx]))
# print('top25 alternatives:', pruned_list[:25])
# assign the token for generation