ai-toolkit/toolkit/data_transfer_object at 70453703467d46904421c1ef9ac8a8cd13e54389 - 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