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