da03 commited on
Commit
005a619
·
1 Parent(s): 7ea045b
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -192,6 +192,7 @@ def _process_frame_sync(model, inputs, use_rnn, num_sampling_steps):
192
 
193
  # UNet sampling
194
  start = time.perf_counter()
 
195
  print (f"USE_RNN: {use_rnn}, NUM_SAMPLING_STEPS: {num_sampling_steps}")
196
  if use_rnn:
197
  sample_latent = output_from_rnn[:, :16]
 
192
 
193
  # UNet sampling
194
  start = time.perf_counter()
195
+ model.clip_denoised = False
196
  print (f"USE_RNN: {use_rnn}, NUM_SAMPLING_STEPS: {num_sampling_steps}")
197
  if use_rnn:
198
  sample_latent = output_from_rnn[:, :16]