From 7c8e969b2ae65bb11ec3bb0c3aced8e790672e50 Mon Sep 17 00:00:00 2001 From: Max Woolf Date: Tue, 18 Jun 2019 22:31:25 -0700 Subject: [PATCH] Remove assertion due to false positives --- gpt_2_simple/gpt_2.py | 1 - setup.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/gpt_2_simple/gpt_2.py b/gpt_2_simple/gpt_2.py index b0e44a8..5c73447 100644 --- a/gpt_2_simple/gpt_2.py +++ b/gpt_2_simple/gpt_2.py @@ -391,7 +391,6 @@ def generate(sess, if prefix: context = tf.placeholder(tf.int32, [batch_size, None]) context_tokens = enc.encode(prefix) - assert len(context_tokens) < length np.random.seed(seed) tf.set_random_seed(seed) diff --git a/setup.py b/setup.py index 1357311..e92d9f9 100644 --- a/setup.py +++ b/setup.py @@ -47,7 +47,7 @@ setup( name='gpt_2_simple', packages=['gpt_2_simple'], # this must be the same as the name above - version='0.5.2', + version='0.5.3', description="Python package to easily retrain OpenAI's GPT-2 " \ "text-generating model on new texts.", long_description=long_description,