2x 6kpro

#1
by mtcl - opened

You said that it fits on it. How much context we can get here please?

This is so exciting. Thank you!

Currently running GLM-5.3-Flash with llama.cpp from unsloth with IQ4_XS gguf. During token generation GPUs are taking turns and load is about 50%, so tg stays on about 50 tok/s without mtp.
So looking recipe for vLLM.

Got it finally running with following commands but it is very tight on VRAM and 32k context does not fit. It is much faster (tg ~80 tok/s without mtp):

Model loading took 86.08 GiB per GPU
Available KV cache memory: 1.3 GiB
GPU KV cache size: 26,437 tokens, Maximum concurrency for 16,384 tokens: 1.61x
Generation: 87-90 tok/s, no speculator

IMAGE=voipmonitor/vllm@sha256:488ddf752938b5ab17e3083dd7d5bb84f418bc3f8856f93cc514c8b66abbe4c6

docker run --name jovian-mtp3-tp2 --init --gpus '"device=0,1"' --network host --ipc host --shm-size 32g -v jovian-judgement-vllm-cache:/cache -v ~/.cache/huggingface:/root/.cache/huggingface -e HF_HUB_OFFLINE=1 -e MODEL=local-inference-lab/GLM-5.3-Flash-NVFP4-4p67 -e PORT=8080 -e HOST=0.0.0.0 -e TP=2 -e DCP=1 -e MAX_MODEL_LEN=16384 -e MAX_NUM_SEQS=4 -e MAX_NUM_BATCHED_TOKENS=1024 -e MAX_CUDAGRAPH_CAPTURE_SIZE=48 -e GPU_MEMORY_UTILIZATION=0.97 -e B12X_PCIE_ALLREDUCE=1 -e NCCL_MIN_NCHANNELS=32 -e NCCL_MAX_NCHANNELS=32 -e NCCL_CUMEM_ENABLE=0 -e NCCL_IB_DISABLE=1 -e NCCL_P2P_LEVEL=SYS -e NCCL_PROTO=LL,LL128,Simple -e OMP_NUM_THREADS=2 -e INSTANTTENSOR_BUFFER_SIZE=67108864 -e INSTANTTENSOR_IO_DEPTH=3 -e INSTANTTENSOR_CONCURRENCY=1 -e INSTANTTENSOR_CHUNK_SIZE=8388608 "$IMAGE" --kv-cache-memory 1400000000

vLLM print following notification: "WARNING kv_cache_utils.py:1253] Add 10 padding layers, may waste at most 29.41% KV cache memory
Keeping split GLM-5.3 cache groups with physical page sizes [287232, 2170880]"

Got it running with 512k context and 2 parallel request using sglang and model :
https://github.com/ormandj/sglang-glm53-flash-sm120
https://huggingface.co/ormandj/GLM-5.3-Flash-W4A16-NVFP4-K32-Experts-FP8-WO

Now it flies 100-200tok/s, both GPUs at 100% when processing

Sign up or log in to comment