Skip to content

Commit 8de7800

Browse files
authored
Add fps argument to export_to_gif function. (huggingface#6786)
1 parent 3ac2357 commit 8de7800

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/utils/export_utils.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def buffered_writer(raw_f):
2828
f.flush()
2929

3030

31-
def export_to_gif(image: List[PIL.Image.Image], output_gif_path: str = None) -> str:
31+
def export_to_gif(image: List[PIL.Image.Image], output_gif_path: str = None, fps: int = 10) -> str:
3232
if output_gif_path is None:
3333
output_gif_path = tempfile.NamedTemporaryFile(suffix=".gif").name
3434

@@ -37,7 +37,7 @@ def export_to_gif(image: List[PIL.Image.Image], output_gif_path: str = None) ->
3737
save_all=True,
3838
append_images=image[1:],
3939
optimize=False,
40-
duration=100,
40+
duration=1000 // fps,
4141
loop=0,
4242
)
4343
return output_gif_path

0 commit comments

Comments
 (0)