recipe.toml 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  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. # Optional DFlash speculative decoding with draft model:
  12. # poolside/Laguna-S-2.1-DFlash-NVFP4
  13. # → 15 speculative tokens, requires --max-num-seqs 32.
  14. #
  15. # REFERENCE:
  16. # https://huggingface.co/poolside/Laguna-S-2.1-NVFP4
  17. engine = "vllm"
  18. name = "Laguna S 2.1 NVFP4 (vLLM)"
  19. docker_image = "ghcr.io/timothystewart6/vllm-gb10:v0.23.0-gb10.1"
  20. model = "poolside/Laguna-S-2.1-NVFP4"
  21. tensor_parallel_size = 1
  22. pipeline_parallel_size = 1
  23. entrypoint = ["vllm", "serve"]
  24. args = """
  25. --gpu-memory-utilization 0.85
  26. --max-model-len 262144
  27. --load-format instanttensor
  28. --enable-prefix-caching
  29. --tool-call-parser poolside_v1
  30. --reasoning-parser poolside_v1
  31. --enable-auto-tool-choice
  32. --trust-remote-code
  33. --kv-cache-dtype fp8
  34. --enable-flashinfer-autotune
  35. --max-num-seqs 32
  36. --override-generation-config '{"temperature":0.7,"top_p":0.95}'
  37. --served-model-name laguna-s-2.1
  38. """
  39. [env]
  40. SAFETENSORS_FAST_GPU = "1"
  41. TORCH_CUDA_ARCH_LIST = "12.1a"
  42. FLASHINFER_CUDA_ARCH_LIST = "12.1a"
  43. VLLM_ALLOW_LONG_MAX_MODEL_LEN = "1"
  44. VLLM_FLOAT32_MATMUL_PRECISION = "high"
  45. PYTORCH_CUDA_ALLOC_CONF = "expandable_segments:True"
  46. MAX_JOBS = "4"