Explicitly set prefix to None if empty

minimaxir · Jun 16, 2019 · e715c8d · e715c8d
1 parent ae3436d
commit e715c8d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gpt_2_simple/gpt_2.py b/gpt_2_simple/gpt_2.py
@@ -375,6 +375,9 @@ def generate(sess,
     if nsamples == 1:
         sample_delim = ''
 
+    if prefix == '':
+        prefix = None
+
     if prefix:
         context = tf.placeholder(tf.int32, [batch_size, None])