snapshot.yaml 1.0 KB

1234567891011121314
  1. # EXPERIMENT: local weight snapshots (weight_snapshot.py). The first boot loads
  2. # the checkpoint and writes one snapshot per rank; later boots load from it.
  3. services:
  4. glm53:
  5. volumes:
  6. - ../experimental/snapshot/weight_snapshot.py:/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/weight_snapshot.py:ro
  7. - ../experimental/snapshot/base_loader.py:/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/base_loader.py:ro
  8. # CUTLASS NVFP4 MoE under weights_already_processed: allow it, and skip the
  9. # in-place scale fold the snapshot already holds.
  10. - ../experimental/snapshot/modelopt.py:/usr/local/lib/python3.12/dist-packages/vllm/model_executor/layers/quantization/modelopt.py:ro
  11. - ../experimental/snapshot/flashinfer_cutlass_moe.py:/usr/local/lib/python3.12/dist-packages/vllm/model_executor/layers/fused_moe/experts/flashinfer_cutlass_moe.py:ro
  12. environment:
  13. - VLLM_WEIGHT_SNAPSHOT_DIR=/root/.cache/weight-snapshots
  14. - VLLM_WEIGHT_SNAPSHOT_TAG=2026-09-28