Skip to content

[diffusers-cli] add a dockerfile for diffusers cli hf jobs - #14550

Open
sayakpaul wants to merge 11 commits into
mainfrom
diffusers-cli-cuda-docker
Open

sayakpaul wants to merge 11 commits into
mainfrom
diffusers-cli-cuda-docker

Conversation

@sayakpaul

@sayakpaul sayakpaul commented Aug 21, 2026 •

Copy link
Copy Markdown
Member

Used this command to launch a job that used the Docker container above:

diffusers-cli --format json run \
    --model black-forest-labs/FLUX.1-dev \
    --dtype bf16 \
    --pipeline-kwargs '{"prompt": "a red panda astronaut floating above a coral reef, cinematic lighting", 
  "num_inference_steps": 20, "guidance_scale": 3.5}' \
    --seed 0 \
    --remote --flavor a100-large --timeout 30m

Resulting job: https://huggingface.co/jobs/sayakpaul/6a87c657c81709766c80327a

Note that the job shows "canceled" rather than "completed". This is because we use .kill() here:

I think this is misleading.

Resulting image:

image

@github-actions github-actions Bot added documentation Improvements or additions to documentation utils CI size/M PR with diff < 200 LOC labels Aug 21, 2026
@HuggingFaceDocBuilderDev

Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@sayakpaul
sayakpaul marked this pull request as ready for review August 21, 2026 03:49
@sayakpaul
sayakpaul requested a review from DN6 August 21, 2026 03:49
Comment thread docker/diffusers-cli-cuda/Dockerfile Outdated
imageio \
imageio-ffmpeg

RUN uv pip install --system --break-system-packages --no-cache-dir "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers"

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I would install from PyPI for a stable version and not from main. If the user needs to use diffusers from main, they can override the install with the --dependencies flag in the CLI, or pass a custom image.

@sayakpaul
sayakpaul requested a review from DN6 August 25, 2026 11:25
@github-project-automation github-project-automation Bot moved this to In Progress in Diffusers Roadmap Sep 3, 2026
@sayakpaul sayakpaul added this to the Release 0.41.0 milestone Sep 23, 2026
@sayakpaul

Copy link
Copy Markdown
Member Author

@DN6 a gentle ping.

fix(ci): harden workflow files flagged on #14550

Co-authored-by: hf-security-analysis[bot] <265538906+hf-security-analysis[bot]@users.noreply.github.com>

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CI documentation Improvements or additions to documentation size/M PR with diff < 200 LOC utils

Projects

Status: In Progress

Development

Successfully merging this pull request may close these issues.

3 participants