Upload verified Muse Spark DSpark step-2000 checkpoint
Browse files- export_manifest.json +7 -7
- model.safetensors +1 -1
- train_command.txt +3 -3
export_manifest.json
CHANGED
|
@@ -1,12 +1,12 @@
|
|
| 1 |
{
|
| 2 |
"format_version": 1,
|
| 3 |
-
"created_at_utc": "2026-09-
|
| 4 |
-
"source_checkpoint": "/home/inf-zixi/speculator-training/muse_spark/training2/scale_all_global_20260909/evaluation/
|
| 5 |
"target_model": "/home/inf-zixi/.cache/huggingface/hub/models--meta-models--Muse-Spark-1.2-820B-A42B-NVFP4/snapshots/b8ace9535a34b9c33a2ef5bfbc86189f56c73e2b",
|
| 6 |
"source_config_sha256": "0cd8fa598e4939499bede4156d40c6abd1e945dc4b1c12bfb8d6442782525b9d",
|
| 7 |
-
"source_weights_sha256": "
|
| 8 |
"output_config_sha256": "588298a4cd6e5179467656644c33d29c31f33f4c951f5c26d2b43aaa390b9af7",
|
| 9 |
-
"output_weights_sha256": "
|
| 10 |
"source_weight_bytes": 7292920442,
|
| 11 |
"output_weight_bytes": 10604891514,
|
| 12 |
"tensor_count": 74,
|
|
@@ -16,8 +16,8 @@
|
|
| 16 |
"lm_head.weight"
|
| 17 |
],
|
| 18 |
"training_state": {
|
| 19 |
-
"epoch":
|
| 20 |
-
"local_step":
|
| 21 |
-
"global_step":
|
| 22 |
}
|
| 23 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"format_version": 1,
|
| 3 |
+
"created_at_utc": "2026-09-10T19:55:48.044685+00:00",
|
| 4 |
+
"source_checkpoint": "/home/inf-zixi/speculator-training/muse_spark/training2/scale_all_global_20260909/evaluation/checkpoint_20260910_195524/native",
|
| 5 |
"target_model": "/home/inf-zixi/.cache/huggingface/hub/models--meta-models--Muse-Spark-1.2-820B-A42B-NVFP4/snapshots/b8ace9535a34b9c33a2ef5bfbc86189f56c73e2b",
|
| 6 |
"source_config_sha256": "0cd8fa598e4939499bede4156d40c6abd1e945dc4b1c12bfb8d6442782525b9d",
|
| 7 |
+
"source_weights_sha256": "6d0aa34c59ff1fc72dcbafb06271e892fa835747fe59dd07e85cba3d6217e759",
|
| 8 |
"output_config_sha256": "588298a4cd6e5179467656644c33d29c31f33f4c951f5c26d2b43aaa390b9af7",
|
| 9 |
+
"output_weights_sha256": "0d6fd02243888d18e1495cc672c299d660849c390ffb6f6d8b872ac7208b4ada",
|
| 10 |
"source_weight_bytes": 7292920442,
|
| 11 |
"output_weight_bytes": 10604891514,
|
| 12 |
"tensor_count": 74,
|
|
|
|
| 16 |
"lm_head.weight"
|
| 17 |
],
|
| 18 |
"training_state": {
|
| 19 |
+
"epoch": 2,
|
| 20 |
+
"local_step": 1584,
|
| 21 |
+
"global_step": 2000
|
| 22 |
}
|
| 23 |
}
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 10604891514
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0d6fd02243888d18e1495cc672c299d660849c390ffb6f6d8b872ac7208b4ada
|
| 3 |
size 10604891514
|
train_command.txt
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
-
# Timestamp: 2026-09-
|
| 2 |
-
# Git SHA:
|
| 3 |
# World size: 16
|
| 4 |
# speculators: 0.8.0.dev226
|
| 5 |
# vllm: 0.27.2rc1.dev107+gd6f17f3d5.precompiled
|
| 6 |
# transformers: 5.14.1
|
| 7 |
# torch: 2.13.0+cu130
|
| 8 |
# compressed-tensors: 0.17.0
|
| 9 |
-
scripts/train.py --speculator-type dspark --block-size 7 --markov-rank 512 --markov-head-type vanilla --enable-confidence-head --confidence-head-with-markov --from-pretrained /home/inf-zixi/speculator-training/muse_spark/training2/pilots_20260909/all_global/initial --verifier-name-or-path /home/inf-zixi/.cache/huggingface/hub/models--meta-models--Muse-Spark-1.2-820B-A42B-NVFP4/snapshots/b8ace9535a34b9c33a2ef5bfbc86189f56c73e2b --target-layer-ids 14 30 46 58 62 --reuse-last-hidden-state-as-aux --draft-vocab-size 202048 --mask-token-id 200018 --draft-attn-impl simple_flex_attention --loss-implementation fused --loss-fn '{"ce": 0.1, "tv": 1.8}' --dflash-decay-gamma 4.0 --per-position-loss-weight fixed-exp-decay --max-anchors 1024 --data-path /home/inf-zixi/speculator-training/muse_spark/training2/pilots_20260909/data --train-data-ratio 0.98 --total-seq-len 32768 --noise-std 0 --num-workers 5 --prefetch-factor 2 --hidden-states-dtype bfloat16 --hidden-states-backend mooncake --mooncake-master 10.78.208.151:50051 --mooncake-metadata-server P2PHANDSHAKE --mooncake-protocol tcp --mooncake-global-segment-size 0 --mooncake-local-buffer-size 8589934592 --vllm-endpoint http://r01-p01-dgx-c01:8000/v1,http://r01-p01-dgx-c02:8000/v1,http://r01-p01-dgx-c03:8000/v1,http://r01-p01-dgx-c05:8000/v1,http://r01-p01-dgx-c06:8000/v1,http://r01-p01-dgx-c07:8000/v1,http://r01-p01-dgx-c08:8000/v1,http://r01-p01-dgx-c09:8000/v1,http://r01-p01-dgx-c10:8000/v1,http://r01-p01-dgx-c11:8000/v1,http://r01-p01-dgx-c12:8000/v1,http://r01-p01-dgx-c13:8000/v1 --on-missing generate --on-generate delete --request-timeout 1800 --max-retries 2 --epochs 4 --gradient-accumulation-steps 8 --optimizer adamw --lr 3e-4 --weight-decay 0 --scheduler-type cosine --scheduler-warmup-ratio 0.05 --scheduler-total-steps 3605 --scheduler-min-lr-ratio 0.1 --max-grad-norm 1.0 --max-nonfinite-grad-skips 0 --fsdp-shard --checkpoint-freq 1 --checkpoint-steps 100 --retain-step-checkpoints --retain-step-checkpoints-keep-latest 4 --retain-step-checkpoints-milestone-interval 500 --log-freq 1 --logger jsonl --seed 42 --run-name muse_training2_all_global_scale16_4epochs --save-path /home/inf-zixi/speculator-training/muse_spark/training2/scale_all_global_20260909/production/checkpoints --log-dir /home/inf-zixi/speculator-training/muse_spark/training2/scale_all_global_20260909/production/logs-
|
|
|
|
| 1 |
+
# Timestamp: 2026-09-10T15:32:19.131341+00:00
|
| 2 |
+
# Git SHA: f71a86e0dbfd1e7801ecc225d9c3ca0423a52863
|
| 3 |
# World size: 16
|
| 4 |
# speculators: 0.8.0.dev226
|
| 5 |
# vllm: 0.27.2rc1.dev107+gd6f17f3d5.precompiled
|
| 6 |
# transformers: 5.14.1
|
| 7 |
# torch: 2.13.0+cu130
|
| 8 |
# compressed-tensors: 0.17.0
|
| 9 |
+
scripts/train.py --speculator-type dspark --block-size 7 --markov-rank 512 --markov-head-type vanilla --enable-confidence-head --confidence-head-with-markov --from-pretrained /home/inf-zixi/speculator-training/muse_spark/training2/pilots_20260909/all_global/initial --verifier-name-or-path /home/inf-zixi/.cache/huggingface/hub/models--meta-models--Muse-Spark-1.2-820B-A42B-NVFP4/snapshots/b8ace9535a34b9c33a2ef5bfbc86189f56c73e2b --target-layer-ids 14 30 46 58 62 --reuse-last-hidden-state-as-aux --draft-vocab-size 202048 --mask-token-id 200018 --draft-attn-impl simple_flex_attention --loss-implementation fused --loss-fn '{"ce": 0.1, "tv": 1.8}' --dflash-decay-gamma 4.0 --per-position-loss-weight fixed-exp-decay --max-anchors 1024 --data-path /home/inf-zixi/speculator-training/muse_spark/training2/pilots_20260909/data --train-data-ratio 0.98 --total-seq-len 32768 --noise-std 0 --num-workers 5 --prefetch-factor 2 --hidden-states-dtype bfloat16 --hidden-states-backend mooncake --mooncake-master 10.78.208.151:50051 --mooncake-metadata-server P2PHANDSHAKE --mooncake-protocol tcp --mooncake-global-segment-size 0 --mooncake-local-buffer-size 8589934592 --vllm-endpoint http://r01-p01-dgx-c01:8000/v1,http://r01-p01-dgx-c02:8000/v1,http://r01-p01-dgx-c03:8000/v1,http://r01-p01-dgx-c05:8000/v1,http://r01-p01-dgx-c06:8000/v1,http://r01-p01-dgx-c07:8000/v1,http://r01-p01-dgx-c08:8000/v1,http://r01-p01-dgx-c09:8000/v1,http://r01-p01-dgx-c10:8000/v1,http://r01-p01-dgx-c11:8000/v1,http://r01-p01-dgx-c12:8000/v1,http://r01-p01-dgx-c13:8000/v1 --on-missing generate --on-generate delete --request-timeout 1800 --max-retries 2 --epochs 4 --gradient-accumulation-steps 8 --optimizer adamw --lr 3e-4 --weight-decay 0 --scheduler-type cosine --scheduler-warmup-ratio 0.05 --scheduler-total-steps 3605 --scheduler-min-lr-ratio 0.1 --max-grad-norm 1.0 --max-nonfinite-grad-skips 0 --fsdp-shard --checkpoint-freq 1 --checkpoint-steps 100 --retain-step-checkpoints --retain-step-checkpoints-keep-latest 4 --retain-step-checkpoints-milestone-interval 500 --log-freq 1 --logger jsonl --seed 42 --run-name muse_training2_all_global_scale16_4epochs --save-path /home/inf-zixi/speculator-training/muse_spark/training2/scale_all_global_20260909/production/checkpoints --log-dir /home/inf-zixi/speculator-training/muse_spark/training2/scale_all_global_20260909/production/logs-attempt3/train
|