w32zhong commited on
Commit
65597fc
·
verified ·
1 Parent(s): 932d75e

Upload specforge_het.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. specforge_het.json +131 -0
specforge_het.json ADDED
@@ -0,0 +1,131 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset.debug": false,
3
+ "dataset.eval_path": "output/datasets/eval_ds_Qwen3-4B-Instruct-2507",
4
+ "dataset.git_diff": "",
5
+ "dataset.git_sha1": "unknown",
6
+ "dataset.manual_sample_ids": [],
7
+ "dataset.max_read_items": null,
8
+ "dataset.output_dir": "output",
9
+ "dataset.path": "./output/datasets/ds_Qwen3-4B-Instruct-2507",
10
+ "dataset.read_eagle_format": false,
11
+ "dataset.run_name": "temp_run",
12
+ "dataset.seed": 0,
13
+ "dataset_generation.batch_size": 1,
14
+ "dataset_generation.debug": false,
15
+ "dataset_generation.debug_target": null,
16
+ "dataset_generation.ds_prefix": "ds_",
17
+ "dataset_generation.git_diff": "",
18
+ "dataset_generation.git_sha1": "unknown",
19
+ "dataset_generation.max_length": 2048,
20
+ "dataset_generation.output_dir": "output",
21
+ "dataset_generation.run_name": "temp_run",
22
+ "dataset_generation.save_every": 1000,
23
+ "dataset_generation.seed": 0,
24
+ "dataset_generation.sharegpt_path": "Aeala/ShareGPT_Vicuna_unfiltered",
25
+ "device_names": [
26
+ "NVIDIA H100 80GB HBM3"
27
+ ],
28
+ "inference.debug": false,
29
+ "inference.dynamic_draft_all_top_k": 59,
30
+ "inference.dynamic_draft_max_depth": 6,
31
+ "inference.dynamic_draft_top_k": 10,
32
+ "inference.git_diff": "",
33
+ "inference.git_sha1": "unknown",
34
+ "inference.max_new_tokens": 2048,
35
+ "inference.output_dir": "output",
36
+ "inference.run_name": "temp_run",
37
+ "inference.seed": 0,
38
+ "inference.timer": false,
39
+ "modeling.chat_template": "models/speculative_qwen3/chat_template.jinja2",
40
+ "modeling.debug": false,
41
+ "modeling.draft_config_custom_modify": {
42
+ "num_experts_per_tok": "2"
43
+ },
44
+ "modeling.draft_config_modify": {
45
+ "moe_intermediate_size": "base.intermediate_size // draft_config.num_experts_per_tok",
46
+ "num_key_value_heads": "base.num_key_value_heads",
47
+ "rope_theta": "base.rope_theta"
48
+ },
49
+ "modeling.dtype": "torch.float32",
50
+ "modeling.free_base_layers": "num_hidden_layers",
51
+ "modeling.git_diff": "",
52
+ "modeling.git_sha1": "unknown",
53
+ "modeling.init_base_model": [
54
+ "SpeculativeQwen3ForCausalLM",
55
+ "Qwen/Qwen3-4B-Instruct-2507"
56
+ ],
57
+ "modeling.init_draft_config": [
58
+ "Qwen3MoeDrafter",
59
+ "Qwen/Qwen3-30B-A3B-Instruct-2507"
60
+ ],
61
+ "modeling.init_speculative_algorithm": [
62
+ "EagleV2",
63
+ "dict(draft_layers=1)"
64
+ ],
65
+ "modeling.max_memory": null,
66
+ "modeling.model_path": null,
67
+ "modeling.output_dir": "output",
68
+ "modeling.run_name": "temp_run",
69
+ "modeling.seed": 0,
70
+ "modeling.stand_alone_draft_model_key_adapt": null,
71
+ "modeling.stand_alone_draft_model_key_modify": {
72
+ "yuhuili": [
73
+ [
74
+ "^fc\\.",
75
+ "eagle_fc."
76
+ ],
77
+ [
78
+ "embed_tokens.weight",
79
+ null
80
+ ]
81
+ ]
82
+ },
83
+ "modeling.stand_alone_draft_model_path": null,
84
+ "modeling.tokenizer_add_tokens": "dict(unk_token=tokenizer.eos_token)",
85
+ "modeling.tokenizer_init": "",
86
+ "modeling.tokenizer_path": "Qwen/Qwen3-4B-Instruct-2507",
87
+ "training.adam_beta1": 0.9,
88
+ "training.adam_beta2": 0.95,
89
+ "training.average_tokens_across_devices": false,
90
+ "training.bf16": true,
91
+ "training.dataloader_drop_last": true,
92
+ "training.ddp_backend": "gloo",
93
+ "training.ddp_find_unused_parameters": false,
94
+ "training.debug": false,
95
+ "training.deepspeed": null,
96
+ "training.eval_steps": 100,
97
+ "training.eval_strategy": "steps",
98
+ "training.filter_out_shorts": false,
99
+ "training.git_diff": "",
100
+ "training.git_sha1": "06c2322f8d37cfe7a2b7e1ec808a523be76337d0",
101
+ "training.gradient_accumulation_steps": 8,
102
+ "training.learning_rate": 3e-05,
103
+ "training.logging_first_step": true,
104
+ "training.logging_steps": 25,
105
+ "training.lr_scheduler_type": "constant_with_warmup",
106
+ "training.max_grad_norm": 0.5,
107
+ "training.max_length": 2048,
108
+ "training.max_steps": -1,
109
+ "training.model_init_ckpt": null,
110
+ "training.num_train_epochs": 10,
111
+ "training.optim": "adamw_torch_fused",
112
+ "training.output_dir": "output/fearless-grass-116",
113
+ "training.overwrite_output_dir": true,
114
+ "training.per_device_eval_batch_size": 1,
115
+ "training.per_device_train_batch_size": 1,
116
+ "training.project": "eagle4",
117
+ "training.report_to": "wandb",
118
+ "training.resume_from_checkpoint": false,
119
+ "training.resume_wandb_runid": null,
120
+ "training.run_name": "fearless-grass-116",
121
+ "training.save_steps": 500,
122
+ "training.save_strategy": "steps",
123
+ "training.save_total_limit": 2,
124
+ "training.seed": 0,
125
+ "training.sequential_loading": false,
126
+ "training.tf32": false,
127
+ "training.use_default_num_items_getter": true,
128
+ "training.use_eagle_pipeline": false,
129
+ "training.warmup_steps": 2000,
130
+ "training.world_size": 1
131
+ }