casts the prepared data to int16 (doesn't help with training memory) 2db9436 winglian commited on Apr 18, 2023