diff options
| author | Rose Hogenson <rhogenson@posteo.net> | 2022-02-15 09:02:56 -0800 |
|---|---|---|
| committer | Rose Hogenson <rhogenson@posteo.net> | 2022-02-15 09:02:56 -0800 |
| commit | a4d6412fdaad7fae011d7103d4e40b01091b112d (patch) | |
| tree | 2da00cb07bd3f8bf459e761e5ed5c5cf9f3c1c89 /ai_dungeon.py | |
| parent | Use gpt2 by default. (diff) | |
| download | gpt-adventure-a4d6412fdaad7fae011d7103d4e40b01091b112d.tar.zst | |
Wrap the prologue.
Sometimes the AI generates really long lines.
Diffstat (limited to 'ai_dungeon.py')
| -rwxr-xr-x | ai_dungeon.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ai_dungeon.py b/ai_dungeon.py index 2b976f6..baac638 100755 --- a/ai_dungeon.py +++ b/ai_dungeon.py @@ -110,12 +110,12 @@ def main() -> None: args = parser.parse_args() model = load_model(args.model) script = pick_flavor() - prologue = generate( + prologue = wrap(generate( model, script, max_new_tokens=20, forced_eos_token_id=model.tokenizer.eos_token_id, - ) + )) print(prologue) prompt = prologue prev_response_length = 0 |