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