Skip to content

Commit 9d1e704

Browse files
committed
slight fix to batch size description
1 parent ed49f03 commit 9d1e704

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

src/generate_unconditional_samples.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,7 @@ def sample_model(
2424
reproduce results
2525
:nsamples=0 : Number of samples to return, if 0, continues to
2626
generate samples indefinately.
27-
:batch_size=1 : Number of batches, model runs nsamples//batch_size
28-
times, each batch run is independent of previous run.
27+
:batch_size=1 : Number of batches (only affects speed/memory).
2928
:length=None : Number of tokens in generated text, if None (default), is
3029
determined by model hyperparameters
3130
:temperature=1 : Float value controlling randomness in boltzmann

src/interactive_conditional_samples.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,8 @@ def interact_model(
2222
:model_name=117M : String, which model to use
2323
:seed=None : Integer seed for random number generators, fix seed to reproduce
2424
results
25-
:nsamples=1 : Number of samples to return
26-
:batch_size=1 : Number of batches, model runs nsamples//batch_size
27-
times, each batch run is independent of previous run.
25+
:nsamples=1 : Number of samples to return total
26+
:batch_size=1 : Number of batches (only affects speed/memory). Must divide nsamples.
2827
:length=None : Number of tokens in generated text, if None (default), is
2928
determined by model hyperparameters
3029
:temperature=1 : Float value controlling randomness in boltzmann

0 commit comments

Comments
 (0)