recipe.toml 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  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. engine = "vllm"
  18. name = "Laguna S 2.1 NVFP4 (vLLM)"
  19. docker_image = "ghcr.io/timothystewart6/vllm-gb10:v0.25.1-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", """
  37. '{"temperature":0.7,"top_p":0.95}'""",
  38. # DFlash speculative decoding with quantization-matched draft
  39. "--speculative-config", """
  40. '{"model":"poolside/Laguna-S-2.1-DFlash-NVFP4","num_speculative_tokens":15,"method":"dflash"}'""",
  41. "--served-model-name", "laguna-s-2.1", "laguna", "default",
  42. ]
  43. [env]
  44. SAFETENSORS_FAST_GPU = "1"
  45. TORCH_CUDA_ARCH_LIST = "12.1a"
  46. FLASHINFER_CUDA_ARCH_LIST = "12.1a"
  47. VLLM_ALLOW_LONG_MAX_MODEL_LEN = "1"
  48. VLLM_FLOAT32_MATMUL_PRECISION = "high"
  49. PYTORCH_CUDA_ALLOC_CONF = "expandable_segments:True"
  50. MAX_JOBS = "4"