Fix auto-casting with SFT sometimes only up-casting keys and not queries

#56

Behavior noticed in adding Phi-3 support to the Axolotl codebase.

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment