* fix use of flex attn kwargs and add support for tiledmlp for GPT-OSS * add logging back * update deps
6.1 KiB
6.1 KiB
* fix use of flex attn kwargs and add support for tiledmlp for GPT-OSS * add logging back * update deps