Which Large Language Models Does Petals Support and How Fast Are They?

Updated

Petals supports Llama 3.1 (up to 405B), Mixtral (8x22B), Falcon (40B+), and BLOOM (176B), and it can run them on consumer-grade GPUs or Google Colab by splitting the model across a peer-to-peer network. For single-batch inference, the project reports up to 6 tokens/sec for Llama 2 (70B) and up to 4 tokens/sec for Falcon (180B) — speeds it describes as enough for chatbots and interactive apps.

Supported models

According to the Petals site, you can generate text with:

  • Llama 3.1 — up to 405B parameters
  • Mixtral — 8x22B
  • Falcon — 40B and above
  • BLOOM — 176B

You can also fine-tune these models for your own tasks. The same page lists Llama 2 (70B) and Falcon (180B) in its performance examples, so those sizes are part of the demonstrated range.

How fast is inference?

Petals reports single-batch inference speeds of:

Model Reported speed
Llama 2 (70B) up to 6 tokens/sec
Falcon (180B) up to 4 tokens/sec

These figures are for single-batch inference. The site frames them as sufficient for chatbots and interactive applications — not for high-throughput serving.

What you can do beyond standard APIs

Petals isn't limited to a fixed generate-text call. The site says you can:

  • employ any fine-tuning and sampling methods
  • execute custom paths through the model
  • see its hidden states

That means you get API-like convenience while keeping the flexibility of PyTorch and 🤗 Transformers.

How the distributed setup works

You load a part of the model, then join a network of people serving its other parts. This BitTorrent-style approach is what lets a consumer-grade GPU or a Colab notebook participate in running models far larger than the local hardware could hold alone.

Practical takeaway

If your goal is interactive chat or a demo with a very large open model, Petals' reported 4–6 tokens/sec range is workable. If you need high-throughput batch inference, the single-batch numbers suggest this isn't the target use case. To try it, the site points to a Colab notebook and GitHub docs.