Fixed more bugs, solved an issue with 16FP used for the GRU. e8bfacb verified noch inator commited on Mar 12, 2025
Fixed some model loading errors in train function c8d039e verified noch inator commited on Mar 11, 2025