The Fastest Way to
Fine-tune and Serve LLMs
Built by AI leaders from Uber, Google, Apple and Amazon. Developed and deployed with the world’s leading organizations.
Fine-tune and Serve 100s of Open-source LLMs
The biggest selection of models at industry-leading pricing
CodeLlama 13B Instruct
Code Llama is a collection of pretrained and fine-tuned generative text models ranging in scale...
Try it for free
Phi 3 4k instruct
The Phi-3-Mini-4K-Instruct is a 3.8B parameters, lightweight, state-of-the-art open model trained...
Try it for free
Llama 3 8B
Meta developed and released the Meta Llama 3 family of large language models (LLMs), a collection...
Try it for free
Mixtral 8x7B v01
The Mixtral-8x7B Large Language Model (LLM) is a pretrained generative Sparse Mixture of Experts...
Try it for free
Bigger Isn’t Always Better
Fine-tune smaller task-specific LLMs that outperform bloated alternatives from commercial vendors. Don’t pay for what you don’t need.
Efficient Fine-Tuning and Serving
Train and deploy task-specific open-source models in record time and under budget.
First-class fine-tuning experience
Predibase offers state-of-the-art fine-tuning techniques out of the box such as quantization, low-rank adaptation, and memory-efficient distributed training to ensure your fine-tuning jobs are fast and efficient—even on commodity GPUs.
The most cost-effective serving infra
With Serverless Fine-Tuned Endpoints and token-based pricing you can stop paying for GPU resources you don’t need. Our unique serving infra–LoRAX–lets you cost-effectively serve many fine-tuned adapters on a single GPU in dedicated deployments.
Your Models, Your Property
Start owning and stop renting your LLMs. The models you build and customize on Predibase are your property, regardless of whether you use the Predibase Cloud and Serverless Fine-Tuned Endpoints or deploy inside your VPC.
The fastest way to fine-tune and deploy any open-source LLM
Fine-tune and serve any open-source LLM. Our proven, scalable infrastructure is available through either serverless fine-tuned endpoints or within your environment’s virtual private cloud.
Try Any Open Source LLM in an Instant
Stop spending hours wrestling with complex model deployments before you’ve even started fine-tuning. Deploy and query the latest open-source pre-trained LLM—like Llama-2, Mistral and Zephyr—so you can test and evaluate the best base model for your use case. Scalable managed infrastructure in your VPC or Predibase cloud enables you to achieve this in minutes with just a few lines of code.
# Deploy an LLM from HuggingFace
pb.deployments.create(
name="my-llama-3-70b-deployment",
description="Deployment of Llama 3 70b in Predibase Cloud",
config=DeploymentConfig(
base_model="llama-3-70b",
)
)
# Prompt the deployed LLM
client = pb.deployments.client("my-llama-3-70b-deployment")
client.generate("Write an algorithm in Java to reverse the words in a string.")
Efficiently Fine-tune Models for Your Task
No more out-of-memory errors or costly training jobs. Fine-tune any open-source LLM on the most readily available GPUs using Predibase’s optimized training system. We automatically apply optimizations such as quantization, low-rank adaptation, and memory-efficient distributed training combined with right-sized compute to ensure your jobs are successfully trained as efficiently as possible.
# Kick off the fine-tune job
adapter = pb.adapters.create(
config=FinetuningConfig(
base_model: "llama-3-70b",
epochs: 3,
learning_rate: 0.0002,
),
dataset=my_dataset,
repo="my_adapter",
description='Fine-tune llama-3-70b with my dataset for my task.',
)
Dynamically Serve Many Fine-tuned LLMs In One Deployment
Our scalable serving infra automatically scales up and down to meet the demands of your production environment. Dynamically serve many fine-tuned LLMs together for over 100x cost reduction versus dedicated deployments with our novel LoRA Exchange (LoRAX) architecture. Load and query them in seconds.
Read more about LoRAX.
# Prompt your fine-tuned adapter instantly
client.generate(
"Write an algorithm in Java to reverse the words in a string.",
adapter_id="my_adapter/3",
)
Built on Proven Open-Source Technology
LoRAX
LoRAX (LoRA eXchange) enables users to serve thousands of fine-tuned LLMs on a single GPU, dramatically reducing the cost of serving without compromising on throughput or latency.
Ludwig
Ludwig is a declarative framework to develop, train, fine-tune, and deploy state-of-the-art deep learning and large language models. Ludwig puts AI in the hands of all engineers without requiring low-level code.
Use Cases
Predibase lets you fine-tune any open-source LLM for your task-specific use case.