How much GPU memory does AI21-Jamba-Mini-1.5 need?

About 126 GB for conversations of up to 8,192 tokens: 103.14 GB of weights, the cache for one full-length conversation and the runtime's own memory (an estimate: its settings are behind its licence).

What is the cheapest GPU to run AI21-Jamba-Mini-1.5?

Right now 8x RTX 3090 at $2.29 an hour; 2x RTX PRO 6000D at $3.28 an hour is the best value for its speed. Prices are live offers from GPU clouds, checked every 15 minutes.

Can AI21-Jamba-Mini-1.5 run on an RTX 4090?

Not on one: it needs about 126 GB and an RTX 4090 has 24 GB. Use a bigger GPU, or a smaller (quantized) version of the model.

Models / AI21-Jamba-Mini-1.5

ai21labs/AI21-Jamba-Mini-1.5

Text generation · served with vLLM · ↓ 1.6K downloads a month · View on Hugging Face

ai21labs/AI21-Jamba-Mini-1.5 needs about 126 GB of GPU memory for conversations of up to 8,192 tokens (an estimate: its settings are behind its licence). It runs on 8x RTX 3090 from $2.29/h right now; best value: 2x RTX PRO 6000D at $3.28/h.

The model's publisher asks users to accept its licence (other). GridPup cannot download this one yet: choose an ungated version below.

Weights: 103 GB
GPU memory needed: about 126 GB (estimate)
Conversation length: 8,192 tokens
Memory · disk: 59.6 GB · 154 GB
Architecture: JambaForCausalLM
Licence: other

Hardware and deploy

The publisher has not let GridPup download this licence-gated model yet. An ungated version of it (above) can be deployed.

Questions

How much GPU memory does AI21-Jamba-Mini-1.5 need?: About 126 GB for conversations of up to 8,192 tokens: 103.14 GB of weights, the cache for one full-length conversation and the runtime's own memory (an estimate: its settings are behind its licence).
What is the cheapest GPU to run AI21-Jamba-Mini-1.5?: Right now 8x RTX 3090 at $2.29 an hour; 2x RTX PRO 6000D at $3.28 an hour is the best value for its speed. Prices are live offers from GPU clouds, checked every 15 minutes.
Can AI21-Jamba-Mini-1.5 run on an RTX 4090?: Not on one: it needs about 126 GB and an RTX 4090 has 24 GB. Use a bigger GPU, or a smaller (quantized) version of the model.

Loading…