xformAI/opt-125m-gqa-ub-6-best-for-KV-cache

来源:🤗text-generationmit1.5 GBother✓ 1 个校验和1,006 天前更新
或提交种子

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 6 groups of KV heads per layer instead of original 12 KV heads in the MHA implementation.