diff --git a/generator.pyc b/generator.pyc new file mode 100644 index 0000000..8e7172d Binary files /dev/null and b/generator.pyc differ diff --git a/gpt2/__pycache__/__init__.cpython-37.pyc b/gpt2/__pycache__/__init__.cpython-37.pyc deleted file mode 100644 index b59cf9e..0000000 Binary files a/gpt2/__pycache__/__init__.cpython-37.pyc and /dev/null differ diff --git a/gpt2/src/__pycache__/encoder.cpython-37.pyc b/gpt2/src/__pycache__/encoder.cpython-37.pyc deleted file mode 100644 index c23a36b..0000000 Binary files a/gpt2/src/__pycache__/encoder.cpython-37.pyc and /dev/null differ diff --git a/gpt2/src/__pycache__/model.cpython-37.pyc b/gpt2/src/__pycache__/model.cpython-37.pyc deleted file mode 100644 index 65b91cf..0000000 Binary files a/gpt2/src/__pycache__/model.cpython-37.pyc and /dev/null differ diff --git a/gpt2/src/__pycache__/sample.cpython-37.pyc b/gpt2/src/__pycache__/sample.cpython-37.pyc deleted file mode 100644 index d243bb1..0000000 Binary files a/gpt2/src/__pycache__/sample.cpython-37.pyc and /dev/null differ diff --git a/main.py b/main.py index 3e17c1d..21dbe04 100644 --- a/main.py +++ b/main.py @@ -135,7 +135,10 @@ def teardown_sess(_): sess.close() def generate_cache(): - + + + generator = StoryGenerator(tf.Session()) + prompt_num = 0 for seed in range(100): result = retrieve_from_cache(seed, prompt_num, [], "story") if result is not None: