da03 commited on
Commit
2f5fac4
·
1 Parent(s): 005a619
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
  model.clip_denoised = False
196
  print (f"USE_RNN: {use_rnn}, NUM_SAMPLING_STEPS: {num_sampling_steps}")
197
  if use_rnn:
 
192
 
193
  # UNet sampling
194
  start = time.perf_counter()
195
+ print (f"model.clip_denoised: {model.clip_denoised}")
196
  model.clip_denoised = False
197
  print (f"USE_RNN: {use_rnn}, NUM_SAMPLING_STEPS: {num_sampling_steps}")
198
  if use_rnn: