feat(stacks): mistral-small-4-heretic drop-in (abliterated NVFP4 backend swap)
Serves the in-house abliterated Mistral Small 4 (heretic NVFP4, vision-intact) under --served-model-name mistral-small-4 on ana-ml2 GPU0:8010 — a true drop-in for the official mistral-small-4 backend. Both litellm entries (mistral-small-4 + mistral-small-4-reasoning) route here with no litellm change. GPU0 fits one mistral-class model, so this is a backend swap, not a co-tenant: bring up after downing the official stack; revert by downing this and up-ing the official. Verified live through the gateway: standard returns clean answers, reasoning populates reasoning_content (the [THINK] split). Checkpoint built per tools/mistral-small4-nvfp4/.
This commit is contained in:
@@ -0,0 +1,24 @@
|
||||
# mistral-small-4-heretic .env — copy to .env on ana-ml2 (gitignored).
|
||||
# Easiest: copy the official stack's live .env and change MISTRAL_CONTAINER_NAME,
|
||||
# so API_KEY (== litellm's VLLM_API_KEY) and HF_TOKEN carry over unchanged —
|
||||
# that key match is what lets litellm authenticate to this backend with no edit.
|
||||
|
||||
# v0.22.0 = the last vLLM before the Mistral multimodal regression (#44911);
|
||||
# verified working vision tower. Matches the official stack's pin.
|
||||
MISTRAL_IMAGE=vllm/vllm-openai:v0.22.0
|
||||
|
||||
# Distinct from the official's vllm-mistral4 so both can exist on disk; only one
|
||||
# runs at a time (shared GPU0 + port 8010).
|
||||
MISTRAL_CONTAINER_NAME=vllm-mistral4-heretic
|
||||
|
||||
# Same port + GPU as the official → true drop-in after the official is downed.
|
||||
MISTRAL_PORT=8010
|
||||
MISTRAL_GPU_ID=0
|
||||
MISTRAL_GPU_MEM_UTIL=0.93
|
||||
MISTRAL_MAX_MODEL_LEN=262144
|
||||
MISTRAL_MAX_NUM_SEQS=32
|
||||
|
||||
# Must equal the official stack's API_KEY (== litellm VLLM_API_KEY env). Carried
|
||||
# over by copying the official .env. Real value lives on the host, gitignored.
|
||||
API_KEY=
|
||||
HF_TOKEN=
|
||||
@@ -0,0 +1,101 @@
|
||||
# mistral-small-4-heretic — abliterated Mistral Small 4 (heretic NVFP4) as a
|
||||
# DROP-IN for the official mistral-small-4 backend.
|
||||
#
|
||||
# Serves darkc0de/Mistral-Small-4-119B-2603-heretic, quantized to NVFP4 in-house
|
||||
# (vision tower kept bf16) and converted to Mistral native format. Built + validated
|
||||
# 2026-06-17 — see tools/mistral-small4-nvfp4/ for the build pipeline.
|
||||
#
|
||||
# WHY a separate stack: GPU0 fits only one mistral-class model (~65-70 GB), so this
|
||||
# is a backend SWAP, not a co-tenant. Bring it up only after downing the official
|
||||
# mistral-small-4 stack. It serves under --served-model-name mistral-small-4 on the
|
||||
# SAME port (8010), so litellm's mistral-small-4 + mistral-small-4-reasoning entries
|
||||
# route here with NO litellm change. Revert = down this, `up -d` the official stack.
|
||||
#
|
||||
# DIFFERENCES vs the official compose (everything else mirrors it for a faithful
|
||||
# drop-in — TP=1, util 0.93, MLA, reasoning + tool-call parsers, prefix caching):
|
||||
# - model is a LOCAL native dir (mounted /model), not an HF id, so it needs the
|
||||
# native loader flags: --config-format/--load-format/--tokenizer-mode mistral.
|
||||
# - distinct container_name (vllm-mistral4-heretic) so it can be staged without
|
||||
# colliding with the official container.
|
||||
|
||||
name: mistral-small-4-heretic
|
||||
|
||||
services:
|
||||
vllm-mistral4-heretic:
|
||||
image: ${MISTRAL_IMAGE}
|
||||
container_name: ${MISTRAL_CONTAINER_NAME}
|
||||
restart: unless-stopped
|
||||
ipc: host
|
||||
ports:
|
||||
- "${MISTRAL_PORT}:8000"
|
||||
volumes:
|
||||
- /tank/aimodels/huggingface:/hfcache
|
||||
# The in-house heretic native NVFP4 checkpoint (read-only).
|
||||
- /tank/aimodels/quant-work/heretic-native-nvfp4:/model:ro
|
||||
environment:
|
||||
- HF_HOME=/hfcache
|
||||
- HF_HUB_CACHE=/hfcache/hub
|
||||
- HUGGING_FACE_HUB_TOKEN=${HF_TOKEN:-}
|
||||
- VLLM_API_KEY=${API_KEY:-}
|
||||
command:
|
||||
- /model
|
||||
# Native Mistral format (params.json + consolidated*.safetensors + tekken).
|
||||
- --config-format
|
||||
- mistral
|
||||
- --load-format
|
||||
- mistral
|
||||
- --tokenizer-mode
|
||||
- mistral
|
||||
# SAME served name as the official → litellm routes here unchanged.
|
||||
- --served-model-name
|
||||
- mistral-small-4
|
||||
- --host
|
||||
- 0.0.0.0
|
||||
- --port
|
||||
- "8000"
|
||||
- --tensor-parallel-size
|
||||
- "1"
|
||||
- --gpu-memory-utilization
|
||||
- ${MISTRAL_GPU_MEM_UTIL}
|
||||
- --max-model-len
|
||||
- ${MISTRAL_MAX_MODEL_LEN}
|
||||
- --attention-backend
|
||||
- TRITON_MLA
|
||||
- --tool-call-parser
|
||||
- mistral
|
||||
- --enable-auto-tool-choice
|
||||
- --reasoning-parser
|
||||
- mistral
|
||||
- --max-num-seqs
|
||||
- ${MISTRAL_MAX_NUM_SEQS}
|
||||
- --dtype
|
||||
- auto
|
||||
- --enable-prefix-caching
|
||||
deploy:
|
||||
resources:
|
||||
reservations:
|
||||
devices:
|
||||
- driver: nvidia
|
||||
device_ids:
|
||||
- "${MISTRAL_GPU_ID}"
|
||||
capabilities:
|
||||
- gpu
|
||||
healthcheck:
|
||||
test: ["CMD", "curl", "-f", "http://localhost:8000/health"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
start_period: 600s
|
||||
networks:
|
||||
- tnet
|
||||
labels:
|
||||
- homepage.group=AI Systems
|
||||
- homepage.name=Mistral Small 4 (heretic NVFP4)
|
||||
- homepage.icon=mdi-creation
|
||||
- homepage.description=Abliterated Mistral-Small-4 (heretic NVFP4) drop-in via vLLM (ana-ml2 GPU 0)
|
||||
- homepage.href=http://10.250.50.54:${MISTRAL_PORT}/docs
|
||||
|
||||
networks:
|
||||
tnet:
|
||||
name: traefik-net
|
||||
external: true
|
||||
Reference in New Issue
Block a user