da03 commited on
Commit
cdf1056
·
1 Parent(s): 2fa0612
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -158,7 +158,7 @@ def _process_frame_sync(model, inputs):
158
 
159
  # UNet sampling
160
  start = time.perf_counter()
161
- use_rnn = False
162
  print (f"use_rnn: {use_rnn}")
163
  if use_rnn:
164
  sample_latent = output_from_rnn[:, :16]
 
158
 
159
  # UNet sampling
160
  start = time.perf_counter()
161
+ use_rnn = True
162
  print (f"use_rnn: {use_rnn}")
163
  if use_rnn:
164
  sample_latent = output_from_rnn[:, :16]