da03 commited on
Commit
78ea6a8
·
1 Parent(s): 9725fa4
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -151,7 +151,7 @@ def _process_frame_sync(model, inputs):
151
  if use_rnn:
152
  sample_latent = output_from_rnn[:, :16]
153
  else:
154
- NUM_SAMPLING_STEPS = 32
155
  if NUM_SAMPLING_STEPS >= 1000:
156
  sample_latent = model.p_sample_loop(cond={'c_concat': output_from_rnn}, shape=[1, *LATENT_DIMS], return_intermediates=False, verbose=True)
157
  else:
 
151
  if use_rnn:
152
  sample_latent = output_from_rnn[:, :16]
153
  else:
154
+ NUM_SAMPLING_STEPS = 8
155
  if NUM_SAMPLING_STEPS >= 1000:
156
  sample_latent = model.p_sample_loop(cond={'c_concat': output_from_rnn}, shape=[1, *LATENT_DIMS], return_intermediates=False, verbose=True)
157
  else: