fix use of flex attn kwargs and add support for tiledmlp for GPT-OSS#3116

winglian merged 3 commits into

mainaxolotl-ai-cloud/axolotl:mainfrom

flex-attn-kwargsaxolotl-ai-cloud/axolotl:flex-attn-kwargsCopy head branch name to clipboard