diff --git a/src/generate_unconditional_samples.py b/src/generate_unconditional_samples.py
index 2199f2b..87e2129 100755
--- a/src/generate_unconditional_samples.py
+++ b/src/generate_unconditional_samples.py
@@ -24,8 +24,7 @@ def sample_model(
      reproduce results
     :nsamples=0 : Number of samples to return, if 0, continues to
      generate samples indefinately.
-    :batch_size=1 : Number of batches, model runs nsamples//batch_size
-     times, each batch run is independent of previous run.
+    :batch_size=1 : Number of batches (only affects speed/memory).
     :length=None : Number of tokens in generated text, if None (default), is
      determined by model hyperparameters
     :temperature=1 : Float value controlling randomness in boltzmann
diff --git a/src/interactive_conditional_samples.py b/src/interactive_conditional_samples.py
index 74b322e..166171a 100755
--- a/src/interactive_conditional_samples.py
+++ b/src/interactive_conditional_samples.py
@@ -22,9 +22,8 @@ def interact_model(
     :model_name=117M : String, which model to use
     :seed=None : Integer seed for random number generators, fix seed to reproduce
      results
-    :nsamples=1 : Number of samples to return
-    :batch_size=1 : Number of batches, model runs nsamples//batch_size
-     times, each batch run is independent of previous run.
+    :nsamples=1 : Number of samples to return total
+    :batch_size=1 : Number of batches (only affects speed/memory).  Must divide nsamples.
     :length=None : Number of tokens in generated text, if None (default), is
      determined by model hyperparameters
     :temperature=1 : Float value controlling randomness in boltzmann