ai-toolkit/toolkit/data_transfer_object at 91bdc6dd1551750d388bdc89531daa3d30d680f2 - 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