diff --git a/docs/compute/pipelines/README.mdx b/docs/compute/pipelines/README.mdx index b6704e565c..5eeed2f4a1 100644 --- a/docs/compute/pipelines/README.mdx +++ b/docs/compute/pipelines/README.mdx @@ -89,6 +89,18 @@ The pipeline fine-tunes the model on a single-GPU instance (the template's compu That's it — you have a fine-tuned model registered in your Clarifai model registry, ready to serve, evaluate, or refine further. +:::tip Python-first alternative + +The LoRA template ships with both a YAML scaffold (`config.yaml`) and a Python DSL form (`dsl.py`). To upload directly from the DSL Python file instead of the scaffold: + +```bash +clarifai pipeline upload dsl.py +``` + +Both paths produce the same pipeline. The DSL form is the recommended starting point if you plan to author your own custom pipelines — see the [Pipeline DSL reference](dsl-reference.md) for the full API (`@step`, `step_ref`, `>>` composition, `base_image`, etc.). + +::: + ### Customize Before Running To override defaults at init time, pass `--set key=value` flags. For example, to fine-tune a different base model and change the number of epochs: diff --git a/docs/compute/pipelines/dsl-reference.md b/docs/compute/pipelines/dsl-reference.md index 4d8df62bde..2d0ecd11a4 100644 --- a/docs/compute/pipelines/dsl-reference.md +++ b/docs/compute/pipelines/dsl-reference.md @@ -56,6 +56,8 @@ def prepare_text(input_text: str) -> str: | `assets` | `list[str]` | Files or directories to bundle alongside the step's code. Paths are resolved relative to the source file containing the step. | | `compute` | `ComputeInfo` | Compute resources for the step (CPU, memory, accelerators). See [Compute Requirements](#compute-requirements). | | `python_version` | `str` | Python version to use inside the step's container. Defaults to `"3.12"`. | +| `base_image` | `Optional[str]` | Custom base Docker image for the step's container — e.g., a CUDA-enabled image for GPU steps. Defaults to Clarifai's standard base image. | +| `platform` | `Optional[str]` | Build platform for the step's container image (e.g., `"linux/amd64"`). Defaults to multi-architecture build. | | `secrets` | `dict[str, str]` | Environment variable name → secret resource path. Secrets are mounted as environment variables at step runtime. | ### Step Input Parameters