QAT release?

#3
by Downtown-Case - opened

This models seems great for its size in the playground!

I like the architecture; the 2 shared experts, in particular, are perfect for hybrid inference.

...But have you considered a QAT release for the community? Even if its just applied the expert FFNs? This would make it more accessible, and make this model stand out; currently only Gemma offers such QAT releases, and none close to this size.

I think QAT requires them to train quantized, which takes more compute. But idk.
This may be the QAT one ¯_(ツ)_/¯

Thinking Machines Lab org

Thanks! I saw that, but it's just the NVFP4 quantization of the trained BF16 model, right? Or is it post-trained?

If it's just quantized, it's going to lose some intelligence compared to the BF16 weights.

I think QAT requires them to train quantized, which takes more compute. But idk.

Yeah, exactly. Ideally it would use the exact training regime of the final model (so realistically only the original model trainer can do it). It does take compute, though much less than what the final finetune should require.

I just thought it would be an interesting option, as currently there are very few large QAT models. The last "big" one to do it, that I'm aware of, is ERNIE 4.5:

https://huggingface.co/baidu/ERNIE-4.5-300B-A47B-2Bits-Paddle

They managed to squeeze it down to 2 bits with minimal loss.

Sign up or log in to comment