-
Notifications
You must be signed in to change notification settings - Fork 26
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fix: process T2I batch sequentially to prevent CUDA out of memory err…
…ors (#66) This commit ensures that batches in the T2I pipeline are processed sequentially. This change is necessary because we currently lack the ability to estimate a GPU's VRAM capacity and manage requests accordingly. * process text-to-image requested image count sequentially * refactor: cleanup sequential images code This commit cleans up the sequential images code a bit. --------- Co-authored-by: Rick Staa <[email protected]>
- Loading branch information
1 parent
c05cccf
commit 8b5cd1e
Showing
1 changed file
with
26 additions
and
27 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters