diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index a19b15e..8d9e3de 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -20,7 +20,7 @@ def loss(labels, logits): class CTRLGenerator(): - def __init__(self, control_code="Horror Text: ", generate_num=80, temperature=0.45): + def __init__(self, control_code="Horror Text: ", generate_num=80, temperature=0.3): self.generate_num=generate_num model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/" @@ -301,5 +301,5 @@ class CTRLGenerator(): token_num += 1 - print("****END DEBUG*****\n\n") + print("\n****END DEBUG*****\n") return self.result_replace(result) diff --git a/story/utils.py b/story/utils.py index 76a4075..cc0d5a9 100644 --- a/story/utils.py +++ b/story/utils.py @@ -125,7 +125,8 @@ second_to_first_mappings = [ ("you", "I"), ("you", "me"), ("you'll", "I'll"), - ("yourself", "myself") + ("yourself", "myself"), + ("you've", "I've") ] def capitalize_helper(string):