Spaces:
Running on Zero
Running on Zero
Disable VAE tile parallelism in workers + post-inference logging
Browse files
app.py
CHANGED
|
@@ -278,11 +278,16 @@ def _generate_impl(input_image, prompt, negative_prompt, guidance_scale, seed, s
|
|
| 278 |
generate_image=True,
|
| 279 |
ref_latents=ref_latents,
|
| 280 |
guidance_scale=ff_args.guidance_scale,
|
|
|
|
|
|
|
|
|
|
| 281 |
).videos[0]
|
|
|
|
| 282 |
|
| 283 |
image = torch.from_numpy(np.array(sample.squeeze(1)))
|
| 284 |
out_path = tempfile.NamedTemporaryFile(delete=False, suffix=".png").name
|
| 285 |
ip.save_image(image, out_path)
|
|
|
|
| 286 |
return out_path
|
| 287 |
|
| 288 |
|
|
|
|
| 278 |
generate_image=True,
|
| 279 |
ref_latents=ref_latents,
|
| 280 |
guidance_scale=ff_args.guidance_scale,
|
| 281 |
+
# tiled VAE decode uses threaded CUDA work and can deadlock
|
| 282 |
+
# in forked ZeroGPU workers; 512x512x1 frame fits VRAM easily
|
| 283 |
+
enable_vae_tile_parallelism=False,
|
| 284 |
).videos[0]
|
| 285 |
+
print("[frameforge] pipe() returned", flush=True)
|
| 286 |
|
| 287 |
image = torch.from_numpy(np.array(sample.squeeze(1)))
|
| 288 |
out_path = tempfile.NamedTemporaryFile(delete=False, suffix=".png").name
|
| 289 |
ip.save_image(image, out_path)
|
| 290 |
+
print(f"[frameforge] saved {out_path}", flush=True)
|
| 291 |
return out_path
|
| 292 |
|
| 293 |
|