liujonathan24 commited on
Commit
02f3aa9
·
verified ·
1 Parent(s): edb295c

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +32 -0
  2. dflash_5layer_mix.pt +2 -2
  3. 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:ee7163f01d0cee06fd8cb45b1c4e3eec1c12a83cd3162068dedf5dc04db0de88
3
- size 6387243259
 
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:51fecc3ee07bd4bd45f62907d8192e555791e235c4ba13b6471e07cb768e05f6
3
- size 903348224
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e9888244ddb7899a30fa4ee577e76dfa2000e9cf8a4a24b0934081bd7b26b6f5
3
+ size 2213173569