MohamedRashad commited on
Commit
432b86b
·
verified ·
1 Parent(s): 5ae67ef

Update diffusers_helper/utils.py

Browse files
Files changed (1) hide show
  1. diffusers_helper/utils.py +1 -1
diffusers_helper/utils.py CHANGED
@@ -276,7 +276,7 @@ def save_bcthw_as_mp4(x, output_filename, fps=10):
276
  x = torch.clamp(x.float(), -1., 1.) * 127.5 + 127.5
277
  x = x.detach().cpu().to(torch.uint8)
278
  x = einops.rearrange(x, '(m n) c t h w -> t (m h) (n w) c', n=per_row)
279
- torchvision.io.write_video(output_filename, x, fps=fps, video_codec='h264', options={'crf': '0'})
280
  return x
281
 
282
 
 
276
  x = torch.clamp(x.float(), -1., 1.) * 127.5 + 127.5
277
  x = x.detach().cpu().to(torch.uint8)
278
  x = einops.rearrange(x, '(m n) c t h w -> t (m h) (n w) c', n=per_row)
279
+ torchvision.io.write_video(output_filename, x, fps=fps, video_codec='libx264', options={'crf': '0'})
280
  return x
281
 
282