recipe.toml 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. # Laguna S 2.1 NVFP4 — 1× DGX Spark (GB10), TP=1.
  2. #
  3. # Model: poolside/Laguna-S-2.1-NVFP4
  4. # → https://huggingface.co/poolside/Laguna-S-2.1-NVFP4
  5. #
  6. # 117.6B total / 8.5B active MoE — NVFP4 quantized (~71 GB stored).
  7. # Fits on a single DGX Spark (128 GB unified memory).
  8. # 48 layers (36 SWA, 12 global), 256+1 experts, 262K context.
  9. # KV cache in FP8.
  10. #
  11. # DFlash speculative decoding with quantization-matched draft model:
  12. # poolside/Laguna-S-2.1-DFlash-NVFP4
  13. # → 15 speculative tokens, method dflash.
  14. #
  15. # REFERENCE:
  16. # https://huggingface.co/poolside/Laguna-S-2.1-NVFP4
  17. #
  18. # DEPLOYMENT: tmux (host venv). The pre-built Docker image
  19. # (ghcr.io/timothystewart6/vllm-gb10:v0.25.1-gb10.1) ships the wrong
  20. # flashinfer version, so Docker is non-functional. The sustainable path
  21. # is a host venv with the exact deps from the model card:
  22. #
  23. # uv venv ~/venvs/vllm025 -p 3.12
  24. # uv pip install vllm==0.25.1 --torch-backend=cu130
  25. # uv pip install flashinfer-python==0.6.15.dev20260712 \
  26. # flashinfer-cubin==0.6.15.dev20260712 \
  27. # flashinfer-jit-cache==0.6.15.dev20260712 \
  28. # --extra-index-url https://flashinfer.ai/whl/nightly/ \
  29. # --extra-index-url https://flashinfer.ai/whl/nightly/cu130/
  30. # hf download poolside/Laguna-S-2.1-NVFP4
  31. # hf download poolside/Laguna-S-2.1-DFlash-NVFP4
  32. #
  33. # Then ensure `vllm` is on PATH (activate the venv or add its bin/ to PATH).
  34. # The entrypoint below (`["vllm", "serve"]`) is used directly by tmux.
  35. engine = "vllm"
  36. name = "Laguna S 2.1 NVFP4 (vLLM)"
  37. docker_image = "ghcr.io/timothystewart6/vllm-gb10:v0.25.1-gb10.1"
  38. model = "poolside/Laguna-S-2.1-NVFP4"
  39. tensor_parallel_size = 1
  40. pipeline_parallel_size = 1
  41. deploy = "tmux"
  42. entrypoint = ["vllm", "serve"]
  43. args = [
  44. "--gpu-memory-utilization", "0.85",
  45. "--max-model-len", "262144",
  46. "--load-format", "instanttensor",
  47. "--enable-prefix-caching",
  48. "--tool-call-parser", "poolside_v1",
  49. "--reasoning-parser", "poolside_v1",
  50. "--enable-auto-tool-choice",
  51. "--trust-remote-code",
  52. "--kv-cache-dtype", "fp8",
  53. "--enable-flashinfer-autotune",
  54. "--max-num-seqs", "32",
  55. "--override-generation-config", """
  56. '{"temperature":0.7,"top_p":0.95}'""",
  57. # DFlash speculative decoding with quantization-matched draft
  58. "--speculative-config", """
  59. '{"model":"poolside/Laguna-S-2.1-DFlash-NVFP4","num_speculative_tokens":15,"method":"dflash"}'""",
  60. "--served-model-name", "laguna-s-2.1", "laguna", "default",
  61. ]
  62. [env]
  63. SAFETENSORS_FAST_GPU = "1"
  64. TORCH_CUDA_ARCH_LIST = "12.1a"
  65. FLASHINFER_CUDA_ARCH_LIST = "12.1a"
  66. CUTE_DSL_ARCH = "sm_121a"
  67. VLLM_ALLOW_LONG_MAX_MODEL_LEN = "1"
  68. VLLM_FLOAT32_MATMUL_PRECISION = "high"
  69. PYTORCH_CUDA_ALLOC_CONF = "expandable_segments:True"
  70. MAX_JOBS = "4"