Add a Diffusers section to Train Models - #2833
Merged
davanstrien merged 1 commit intoOct 2, 2026
Merged
Conversation
Runs the FLUX.2 [klein] 4B DreamBooth LoRA script from its GitHub URL (PEP 723 header) on a10g-small; tested on Jobs. Also adds Diffusers to the push_to_hub / output_dir lists in "After it ends". Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01Xu83DsDNs5aqjdh3WYGXHK
|
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
approved these changes
Oct 1, 2026
Member
100%, let's track that @sayakpaul? |
julien-c
approved these changes
Oct 1, 2026
Member
Author
|
Opened huggingface/diffusers#14924 to track this. |
Member
Author
|
Opened an issue to track this on Diffusers side here: huggingface/diffusers#14924 |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Adds a Diffusers section to Train Models (
jobs-training.md), after Axolotl. This was left out of #2795 to be added later.Most of the DreamBooth LoRA scripts in the Diffusers repo declare their dependencies in a script header, so they run on Jobs straight from their GitHub URL, like the Transformers scripts. The example trains a LoRA for FLUX.2 [klein] 4B (Apache-2.0, not gated) on
diffusers/dog-example, capped to 20 steps, and takes about five minutes ona10g-small. The flags follow the README's klein command, except for these:--dataset_namein place of a local folder--do_fp8_training: it needstorchao, which isn't in the header, and a GPU with compute capability 8.9 or higher. The A10G is 8.6.The command was run on Jobs as written.
Also in this PR: Diffusers joins the
--push_to_huband--output_dirlists in "After it ends".Follow-up: it would be nice for the Diffusers docs to have a Jobs guide too, as Axolotl's and TRL's do. This section would then link to it, as the other library sections do.
🤖 Generated with Claude Code
Note
Low Risk
Documentation-only changes with no runtime, API, or security impact.
Overview
Adds Diffusers to the Train Models on Jobs guide: a new section after Axolotl documents running DreamBooth LoRA scripts via
hf jobs uv runfrom GitHub (example: FLUX.2 klein 4B ondiffusers/dog-example, ~5 minutes ona10g-small), including notes on# /// scriptheaders vs--withfor scripts without them.The After it ends section now lists Diffusers alongside Transformers and TRL for
--push_to_huband--output_dir, so post-job Hub upload and bucket checkpoint guidance applies to diffusion training runs too.Reviewed by Cursor Bugbot for commit c4bd337. Bugbot is set up for automated code reviews on this repo. Configure here.