4 Replies
turns out i had to use pytourch2.2.0
i am running kohya_ss flux branch for the reference
but now i get this

it doesn't prevent the training or anything tho, should i worry about it?
Update: Getting average loss NaN now, before it wasn't even with the error
just use template with cuda12.4