init commit

README.md

@@ -1,3 +1,16 @@
 ---
 license: apache-2.0
+library_name: transformers
+base_model:
+- Qwen/QwQ-32B
+base_model_relation: "adapter"
 ---
+
+
+
+This repo contains the AttnGates' weights for QwQ-32B Model introduced by SeerAttention.
+
+[SeerAttention](https://arxiv.org/pdf/2410.13276) introduces learnable AttnGate modules to accelerate the computationally intensive prefill stage of long-context large language models (LLMs) via dynamic block-level sparsity. The AttnGates are trained in a parameter-efficient self-distillation framework, where they learn to mimic the 2D max-pooled attention patterns of the original frozen model, preserving its integrity while avoiding costly retraining. During inference, these gates generate block-sparse binary masks by applying threshold/TopK to their learned soft scores, enabling efficient computation through a custom block-sparse FlashAttention kernel. 
+
+## Original Github Repo
+[https://github.com/microsoft/SeerAttention](https://github.com/microsoft/SeerAttention). 

attn_gate_weights.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d63e9047219819ec64bded486dc2d2c5a51a4127004973c311eaeec276cc6363
+size 251668686

config.json

@@ -0,0 +1,35 @@
+{
+  "_attn_implementation_autoset": true,
+  "_name_or_path": "Qwen/QwQ-32B",
+  "architectures": [
+    "Qwen2ForCausalLM"
+  ],
+  "attention_dropout": 0.0,
+  "base_model": "Qwen/QwQ-32B",
+  "bos_token_id": 151643,
+  "eos_token_id": 151645,
+  "hidden_act": "silu",
+  "hidden_size": 5120,
+  "initializer_range": 0.02,
+  "intermediate_size": 27648,
+  "max_position_embeddings": 131072,
+  "max_window_layers": 64,
+  "model_type": "qwen2",
+  "num_attention_heads": 40,
+  "num_hidden_layers": 64,
+  "num_key_value_heads": 8,
+  "rms_norm_eps": 1e-05,
+  "rope_scaling": null,
+  "rope_theta": 1000000.0,
+  "seerattn_gate_block_size": 64,
+  "seerattn_gate_force_double": false,
+  "seerattn_gate_hidden_size": 128,
+  "seerattn_gate_type": "Qavg_Kmaxminavg",
+  "sliding_window": 32768,
+  "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.49.0",
+  "use_cache": true,
+  "use_sliding_window": false,
+  "vocab_size": 152064
+}