Remove assertion due to false positives

minimaxir · Jun 19, 2019 · 7c8e969 · 7c8e969
1 parent d9b673e
commit 7c8e969
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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,