ai-toolkit/toolkit/data_transfer_object at 534d462ff622e334e1a65b42884d96d98315d75a - 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