Get started building models even faster by using Public Checkpoints.
How It Works
Public checkpoints are a collection of popular public domain machine learning checkpoint files that are loaded by proxiML. If you are planning to use one of the below checkpoints in your model, be sure to select it in the job form as instructed below instead of provisioning worker storage and downloading it yourself.
If you attach a public checkpoint to your job, you accept the terms and conditions documented in any license, readme, or model card file present in the mounted checkpoint directory or at the associated URL listed below.
Currently, the following models are available to attach to jobs for free:
File-based Checkpoints:
stable-diffusion-v1-4
: https://huggingface.co/CompVis/stable-diffusion-v-1-4-originalstable-diffusion-v1-4-full-ema
: https://huggingface.co/CompVis/stable-diffusion-v-1-4-originalstable-diffusion-v1-5
: https://huggingface.co/runwayml/stable-diffusion-v1-5stable-diffusion-v1-5-emaonly
: https://huggingface.co/runwayml/stable-diffusion-v1-5stable-diffusion-v1-5-inpainting
: https://huggingface.co/runwayml/stable-diffusion-inpaintingstable-diffusion-v2
: https://huggingface.co/stabilityai/stable-diffusion-2stable-diffusion-v2-base
: https://huggingface.co/stabilityai/stable-diffusion-2-basestable-diffusion-v2-depth
: https://huggingface.co/stabilityai/stable-diffusion-2-depthstable-diffusion-v2-inpainting
: https://huggingface.co/stabilityai/stable-diffusion-2-inpaintingstable-diffusion-v2-upscaler
: https://huggingface.co/stabilityai/stable-diffusion-x4-upscalerstable-diffusion-v2-1
: https://huggingface.co/stabilityai/stable-diffusion-2-1stable-diffusion-v2-1-base
: https://huggingface.co/stabilityai/stable-diffusion-2-1-base
Diffusers Compatible Checkpoints:
stable-diffusion-v1-4-diffuser
: https://huggingface.co/CompVis/stable-diffusion-v1-4stable-diffusion-v1-5-diffuser
: https://huggingface.co/runwayml/stable-diffusion-v1-5/tree/mainstable-diffusion-v2-diffuser
: https://huggingface.co/stabilityai/stable-diffusion-2stable-diffusion-v2-1-diffuser
: https://huggingface.co/stabilityai/stable-diffusion-2-1
If you would like a public checkpoint added, please contact us with a link to the checkpoint and a brief description of what you need it for.
Using the Web Platform
To select a public checkpoint on the job form, check the public
box when adding a new checkpoint. The checkpoint field contents will automatically update to the list of available public checkpoints.
Using the SDK
To use a public checkpoint, specify the checkpoint as a dictionary with the public
property set to True
.
job = await proximl.jobs.create(
...
model=dict(
...
checkpoints=[
dict(id="stable-diffusion-v2-1", public=True),
"my-checkpoint"
],
)
)