akhaliq HF Staff commited on
Commit
99affe4
·
verified ·
1 Parent(s): 6e13574

Disable VAE tile parallelism in workers + post-inference logging

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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