diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 3de5673..93c9646 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -207,7 +207,7 @@ class CTRLGenerator(): # Make sure only a possible verb is chosen. if first_token: for word in get_possible_verbs(): - prompt_logits[_token][self.word2idx[word]] += 0.1 + prompt_logits[_token][self.word2idx[word]] *= 1000 first_token = False # compute probabilities from logits diff --git a/story/utils.py b/story/utils.py index 7e1509d..0ab5b59 100644 --- a/story/utils.py +++ b/story/utils.py @@ -90,7 +90,7 @@ first_to_second_mappings = [ (" me", " you"), (" us ", " you "), (" us.", " you."), - ("our ", "your ") + (" our", " your") ] second_to_first_mappings = [ @@ -134,7 +134,7 @@ def second_to_first_person(text): return capitalize_first_letters(text) -possible_verbs = ["go", "run", "open", "look", "walk", "make", "try", "say", "tell", "attack", "use", "turn", "fight", "scream", "yell"] +possible_verbs = ["ask", "go", "run", "open", "look", "walk", "make", "try", "say", "tell", "attack", "use", "turn", "fight", "scream", "yell"] def get_possible_verbs(): return possible_verbs