Upload folder using huggingface_hub
Browse files- README.md +32 -0
- dflash_5layer_mix.pt +2 -2
- flm_5layer_uni.pt +2 -2
README.md
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Qwen3-8B Speculators (DFlash & FLM)
|
| 2 |
+
|
| 3 |
+
Speculator models for the [Qwen/Qwen3-8B](https://huggingface.co/Qwen/Qwen3-8B) target model.
|
| 4 |
+
|
| 5 |
+
## Architectures
|
| 6 |
+
|
| 7 |
+
| Model File | `mlp_type` | Draft Layers | Style |
|
| 8 |
+
|------------|------------|--------------|-------|
|
| 9 |
+
| `dflash_5layer_mix.pt` | `dflash` | 5 | DFlash-style |
|
| 10 |
+
| `flm_5layer_uni.pt` | `flm` | 5 | FLM-style |
|
| 11 |
+
|
| 12 |
+
## Training Data
|
| 13 |
+
|
| 14 |
+
These models were trained on the **regenerated dataset** (DFlash-style) derived from Qwen3-8B's own responses to the following sources:
|
| 15 |
+
- **Nemotron Post-Training V2**
|
| 16 |
+
- **CodeAlpaca**
|
| 17 |
+
- **ShareGPT**
|
| 18 |
+
- **UltraChat-200K**
|
| 19 |
+
|
| 20 |
+
Specifically:
|
| 21 |
+
- **dflash_5layer_mix**: Uses a mixture of datasets and a custom Tau schedule (30% pure noise, 20% grid-aligned, 50% uniform).
|
| 22 |
+
- **flm_5layer_uni**: Uses a uniform Tau schedule and standard dataset mixture.
|
| 23 |
+
|
| 24 |
+
## Details
|
| 25 |
+
|
| 26 |
+
- **Target Model**: Qwen/Qwen3-8B
|
| 27 |
+
- **Sequence Length**: 2048
|
| 28 |
+
- **Batch Size**: 5
|
| 29 |
+
- **Draft Layers**: 5 (despite some folder naming suggesting 4, internal config confirms 5)
|
| 30 |
+
|
| 31 |
+
---
|
| 32 |
+
*Trained at Princeton AILab.*
|
dflash_5layer_mix.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7f9bcf259f407d11f50e15854ee1fe02e69380ddf6ff71861f82ae82f273bf5e
|
| 3 |
+
size 2129081431
|
flm_5layer_uni.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e9888244ddb7899a30fa4ee577e76dfa2000e9cf8a4a24b0934081bd7b26b6f5
|
| 3 |
+
size 2213173569
|