search

jaketae / storyteller

Multimodal AI Story Teller, built with Stable Diffusion, GPT, and neural text-to-speech
MIT License
482 stars 64 forks source link

[TODO] Support batch image generation #9

Closed jaketae closed 1 year ago

jaketae commented 1 year ago

Context

Currently, StoryTeller.generate() generates one image at a time. This is for ease of implementation: we generate one image, a subtitle file, and an audio file, then stitch them together to create one frame.

Solution

To speed up generation on GPUs, support batch generation. This is simple with HF's pipeline:

images = self.painter([prompt1, prompt2, ...])
stale[bot] commented 1 year ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.