-1

app.py

@@ -749,17 +749,13 @@ def worker_video(input_video, prompts, n_prompt, seed, batch, resolution, total_
                             split_idx = 1
                         if clean_latents_4x.shape[2] < 2:  # 20250507 pftq: edge case for <=1 sec videos
                             print("£££££££££££££££££££££££££££££££££££ clean_latents_4x")
-                            clean_latents_4x_shape = clean_latents_4x.shape
-                            clean_latents_4x_shape[2] = 2
-                            clean_latents_4x = clean_latents_4x.expand(clean_latents_4x_shape)
+                            clean_latents_4x = clean_latents_4x.expand(-1, -1, 2, -1, -1)
                             
                         if num_2x_frames > 0 and split_idx < len(splits):
                             clean_latents_2x = splits[split_idx]
                             if clean_latents_2x.shape[2] < 2:  # 20250507 pftq: edge case for <=1 sec videos
                                 print("£££££££££££££££££££££££££££££££££££ clean_latents_2x")
-                                clean_latents_2x_shape = clean_latents_2x.shape
-                                clean_latents_2x_shape[2] = 2
-                                clean_latents_2x = clean_latents_2x.expand(clean_latents_2x_shape)
+                                clean_latents_2x = clean_latents_2x.expand(-1, -1, 2, -1, -1)
                             split_idx += 1
                         elif clean_latents_2x.shape[2] < 2:  # 20250507 pftq: edge case for <=1 sec videos
                             clean_latents_2x = clean_latents_4x