[tests] fix anyflow tests (#13855)
* fix anyflow tests
* [tests] fix anyflow tests layerwise casting (#13863)
Fix AnyFlow FAR causal transformer training layerwise / mixed precision tests
---------
Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com>