ai-toolkit/toolkit/data_transfer_object at e9c72b762748e4d31837fbeb3b87d01a5b143321 - ai-toolkit - Gitea: Git with a cup of tea

evan/ai-toolkit

0 Fork 0

Files

T

History

Jaret Burkett 7045370346 Keep cached pixel tensors as uint8 in the dataloader workers and only materialize to fp16 on the main thread

2026-10-14 13:40:49 -06:00

..

data_loader.py

Keep cached pixel tensors as uint8 in the dataloader workers and only materialize to fp16 on the main thread

2026-10-14 13:40:49 -06:00