Docker Model Runner Pull

Docker Model Runner Pull

Certified

Pull a model via Docker Model Runner

Pulls a model from a registry using the Docker Model Runner (DMR) REST API. The model is streamed line by line; each status line is logged at debug level, and a summary is logged at info level once the pull completes. Throws if any line contains an error field or if the server returns a non-2xx response.

yaml
type: io.kestra.plugin.docker.model.Pull

Pull a model with Docker Model Runner

yaml
id: docker_model_pull
namespace: company.team

tasks:
  - id: pull
    type: io.kestra.plugin.docker.model.Pull
    model: ai/smollm2
Properties

Model identifier

The model to pull, e.g. ai/smollm2 or hf.co/org/repo.

Assets this task consumes as inputs or produces as outputs, for lineage tracking and the asset graph (Enterprise Edition). A flow declaring this property on a task is rejected in the open-source edition.

Definitions
assetFailureBehaviorstring
Possible Values
IGNOREFAILWARN

Asset failure behavior

Behavior applied to the task state when a declared asset fails to render, emit, or be persisted (e.g. a lock conflict): FAIL escalates it to FAILED, WARN (default) warns it if it would otherwise succeed, IGNORE leaves the state untouched.

enableAutobooleanstring

Whether to auto-register assets referenced dynamically at runtime that are not statically declared in inputs or outputs.

inputsarray

The assets consumed as inputs.

id*string
Min length1
typestring
outputs

The assets produced as outputs.

id*string
Min length1
Max length150
type*object
descriptionstring
displayNamestring
metadataobject
Default{}
namespacestring
Min length1
Max length150
id*string
Min length1
Max length150
type*object
descriptionstring
displayNamestring
metadataobject
Default{}
namespacestring
Min length1
Max length150
id*string
Min length1
Max length150
type*object
descriptionstring
displayNamestring
metadataobject
Default{}
namespacestring
Min length1
Max length150
id*string
Min length1
Max length150
type*object
descriptionstring
displayNamestring
metadataobject
Default{}
namespacestring
Min length1
Max length150
id*string
Min length1
Max length150
type*object
descriptionstring
displayNamestring
metadataobject
Default{}
namespacestring
Min length1
Max length150
id*string
Min length1
Max length150
type*string
Min length1

Custom asset type

descriptionstring
displayNamestring
metadataobject
Default{}
namespacestring
Min length1
Max length150
Defaulthttp://localhost:12434

Docker Model Runner host

Base URL of the Docker Model Runner REST API. Override when DMR is exposed on a non-default address.