Add `keep_in_memory` option to ImageCaptionJsonlDataset

invoke-ai / invoke-training

https://invoke-ai.github.io/invoke-training/

Apache License 2.0

120 stars 14 forks source link

Closed RyanJDick closed 7 months ago

RyanJDick commented 7 months ago

Add keep_in_memory option to ImageCaptionJsonlDataset. This improves data loading performance for small datasets that can be held in RAM.