Downloads · 30 days
0
xformAI/opt-6.7b-ub-16-qcqa-best-for-KV-cache
opt-6.7b-ub-16-qcqa-best-for-KV-cache is a machine learning model from xformAI. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
This is a QCQA version of the original model facebook/opt-125m. In this version, the original MHA architecture is preserved but instead of having a single K/V head, different K/V heads corresponding to the same group…
Downloads · 30 days
0
Access
Public
Updated Jan 24, 2024
Repo size
—
Likes
14
Public
Click a slice to open those files.
Other1.5 KB · 74%
From the Hugging Face model README
This is a QCQA version of the original model facebook/opt-125m. In this version, the original MHA architecture is preserved but instead of having a single K/V head, different K/V heads corresponding to the same group have the same mean-pooled K or V values. It has upto 16 groups of KV heads per layer instead of the original 32 KV heads in the MHA implementation. This implementation is supposed to more efficient than corresponding GQA one. This has been optimized for KV-cache.