初始化项目,由ModelHub XC社区提供模型

Model: xformAI/opt-125m-gqa-ub-6-best-for-KV-cache
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-06-25 14:56:38 +08:00
commit fe2ec9edaf
9 changed files with 100423 additions and 0 deletions

8
README.md Normal file
View File

@@ -0,0 +1,8 @@
---
license: mit
language:
- en
library_name: transformers
---
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.