Downloads · 30 days
0
xformAI/opt-6.7b-ub-16-gqa
opt-6.7b-ub-16-gqa 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 GQA 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 h…
Downloads · 30 days
0
Access
Public
Updated Jan 24, 2024
Repo size
—
Likes
0
Public
Click a slice to open those files.
Other1.5 KB · 78%
From the Hugging Face model README
This is a GQA 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 16 groups of KV heads per layer instead of original 32 KV heads in the MHA implementation.