DocsInferenceServe custom modelsExamples
Inference · Serve custom models

Examples

Ready-to-run predictor jobs — segmentation, image generation, TTS, classification.

The examples repository

examples/inference_jobs/predict holds complete predictor-mode jobs across modalities. Like the vLLM set, the models are general-purpose tools with no sector attached; each folder is a working reference for one shape of the predict(data) → dict contract.

FolderWhat it servesMin VRAM
sam2.1_segmentationpoint/box-prompted image segmentation — masks out6 GB
FLUX.1-schnell_imagegentext-to-image generation16 GB
Orpheus-TTS-3B_ttstext-to-speech8 GB
image_classificationa compact CV classifier — the smallest real-model reference
minimal_echothe five-line Echo predictor — the contract with nothing else in the waynone (CPU)

Every folder has the same layout:

inference.yaml

module + class_name + init_args + cluster sizing

scripts/<name>_serve.py

the predictor class

submit.py

deploys via the SDK

predict.py

a working client for that job's request shape

Run any of them

i
Or paste the serve script and yaml into the dashboard's inference wizard. Jobs that pull weights from HuggingFace Hub take a few minutes on first boot; the submit script waits for RUNNING.