.

main.py

@@ -512,7 +512,7 @@ async def websocket_endpoint(websocket: WebSocket):
                 #    print ('predicting', f"record_10003/image_{117+len(previous_frames)}.png")
                 print ('previous_actions', previous_actions)
                 next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
-                feedback = False
+                feedback = True
                 if feedback:
                     previous_frames.append(next_frame_append)
                 else:

utils.py

@@ -82,7 +82,7 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
         if DDPM:
             samples_ddim = model.p_sample_loop(cond=c, shape=[1, 4, 48, 64], return_intermediates=False, verbose=True)
         else:
-            samples_ddim, _ = sampler.sample(S=4,
+            samples_ddim, _ = sampler.sample(S=8,
                                          conditioning=c,
                                          batch_size=1,
                                          shape=[4, 48, 64],