Skip to content

Supported Models

Lamini On-demand supports CausalLM models from Hugging Face for tuning and inference, including Llama 3.1, Mistral 3, Phi-3, Qwen 2, and many more.

Lamini Reserved and Self-managed supports more models. Please contact us if you need them.

Note: Flash Attention 2 and 3 are not yet supported.

Model size

With Memory Tuning you can achieve very high factual accuracy with 8B models, without giving up fluent generalization. Using smaller models lowers operating costs and improves latency.

Some factors to consider when thinking about model size:

  • The more active parameters a model has, the more GPU memory is required to use the model.
  • If a model is larger than a single GPU's memory, it needs to run across multiple GPUs. This means exchanging more data across the network, and both inference and tuning will take longer.
  • Tuning requires significantly more GPU memory than inference.

Model loading

The most popular models are preloaded on Lamini On-demand. Other models will be loaded from Hugging Face when requested in an inference or tuning call. Because models are large (usually tens of GBs), downloading them from Hugging Face and then loading them into GPU memory takes time.

Please allow 20-30 minutes for the model to load. Requests for models that have not yet loaded will return an error.

We recommend focusing development on one model or a small set of models, and preloading them. We've seen the highest accuracy and performance gains come from improving data quality and tuning recipes, rather than testing many models hoping to find one that works significantly better out of the box.