Skip to content
English

Switch theme

// Yi (01.AI) · Text LLM

Yi-1.5-34B-Chat-16K: VRAM & local deployment

34.4B parameters. About 25.4 GB at 4-bit and 78.2 GB at BF16 with one 8K-token request.

released 2024-05-15licence: Commercial OKmemory figures are estimates

01Specs

Publisher01.AI · 01-ai
Family / seriesYi (01.AI) · Yi-1.5-16K
TypeText LLM (text-generation)
Parameters34.4B Source: counted from official safetensors weights
Context length16K tokens (16,384)max_position_embeddings in config.json
Architecture60 layers · GQA
Released weightsBF16 · 68.8 GBdownload
LicenceApache-2.0 · Commercial OKPermissive licence (Apache-2.0 / MIT / CC-BY / OpenMDW): commercial use allowed.
Released2024-05-15 (Hugging Face repo created)

02How much VRAM it needs (estimate)

Total = weights + 10% + 0.5GB overhead + KV cache for one request (FP16 KV). More concurrent users multiply the KV column.

PrecisionWeightsTotal @ 8KSmallest GPUMac
BF16 / FP16 (original)68.8 GB 78.2 GB RTX PRO 6000 96GBMac 128GB
FP8 / INT8 / Q8_036.5 GB 42.7 GB 2 × RTX 5090 (64GB)Mac 64GB
4-bit (Q4_K_M / AWQ / GPTQ-Int4)20.9 GB 25.4 GB RTX 5090 32GBMac 64GB

KV cache per request: 8K 2.0 GB

Bits per weight: BF16 16, FP8 8.5, 4-bit 4.85. “Fits” = total ≤ 90% of memory; Mac GPU can use ~2/3 of unified memory (≤36GB) or ~3/4 (larger). For MoE, all experts must be in memory, so size by total parameters. Method →

03How to run it

Only runtimes the publisher or the runtime’s official library names for this model.

Which inference engine to choose →

04Official repositories & quantised builds

1 official repo(s) from 01-ai.

05Other Yi (01.AI) models

All 9 →
ModelParams4-bit totalReleased
Yi-1.5-9B-Chat-16K · same series8.83B7.2 GB2024-05-15
Yi-Coder-1.5B-Chat1.48B3.1 GB2024-08-21
Yi-Coder-9B-Chat8.83B7.2 GB2024-08-21
Yi-1.5-9B-32K8.83B7.2 GB2024-05-15
Yi-1.5-34B-32K34.4B25.4 GB2024-05-15
Yi-1.5-6B6.06B5.1 GB2024-05-11
Yi-1.5-9B-Chat8.83B7.2 GB2024-05-10
Yi-1.5-34B-Chat34.4B25.4 GB2024-05-10

06Sources

Data checked 2024-06-26. Parameters, context and licence are copied from the official repository; memory figures are our estimates.

Data updated: Self-host open models(UTC+8)