Spaces:
Runtime error
Runtime error
da03
commited on
Commit
·
78ea6a8
1
Parent(s):
9725fa4
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 =
|
| 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:
|