From 8de78001df95a641bf6ef942bee9553921d44490 Mon Sep 17 00:00:00 2001 From: Masamune Ishihara <1396267+masaishi@users.noreply.github.com> Date: Thu, 8 Feb 2024 08:29:51 -0800 Subject: [PATCH] Add fps argument to export_to_gif function. (#6786) --- src/diffusers/utils/export_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/utils/export_utils.py b/src/diffusers/utils/export_utils.py index 4cecc2bfdf..24cce07ab8 100644 --- a/src/diffusers/utils/export_utils.py +++ b/src/diffusers/utils/export_utils.py @@ -28,7 +28,7 @@ def buffered_writer(raw_f): f.flush() -def export_to_gif(image: List[PIL.Image.Image], output_gif_path: str = None) -> str: +def export_to_gif(image: List[PIL.Image.Image], output_gif_path: str = None, fps: int = 10) -> str: if output_gif_path is None: output_gif_path = tempfile.NamedTemporaryFile(suffix=".gif").name @@ -37,7 +37,7 @@ def export_to_gif(image: List[PIL.Image.Image], output_gif_path: str = None) -> save_all=True, append_images=image[1:], optimize=False, - duration=100, + duration=1000 // fps, loop=0, ) return output_gif_path