.

main.py

@@ -135,6 +135,8 @@ def draw_trace(image: np.ndarray, previous_actions: List[Tuple[str, List[int]]],
             y_current = y_current *8
         print ('x_current, y_current', x_current, y_current)
         draw.ellipse([x_current-3, y_current-3, x_current+3, y_current+3], fill=(0, 255, 0))
+    else:
+        assert False
     
     return np.array(pil_image)
 

utils.py

@@ -77,8 +77,8 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
         if DEBUG:
             print ('samples_ddim.shape', samples_ddim.shape)
             x_samples_ddim = samples_ddim[:, :3]
-            # upsample to 512 x 368
-            x_samples_ddim = torch.nn.functional.interpolate(x_samples_ddim, size=(512, 368), mode='bilinear')
+            # upsample to 512 x 384
+            x_samples_ddim = torch.nn.functional.interpolate(x_samples_ddim, size=(384, 512), mode='bilinear')
         else:
             x_samples_ddim = model.decode_first_stage(samples_ddim)
         #x_samples_ddim = pos_map.to(c['c_concat'].device).unsqueeze(0).expand(-1, 3, -1, -1)