From 67a7358ee8a9d001d0e21ae726edf59307348556 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?apolin=C3=A1rio?= Date: Thu, 31 Jul 2025 17:17:02 +0200 Subject: [PATCH 1/4] Add UV script comment block This PR makes our advanced diffusion training compatible with uv, which therefore also makes them compatible with Hugging Face jobs: https://huggingface.co/docs/huggingface_hub/en/guides/jobs The idea is to test it out with this PR, working well we could expand the uv comment for all diffusers scripts to add both uv and hf jobs compatibility --- .../train_dreambooth_lora_flux_advanced.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py b/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py index c18d4553ed1c..4150c66c3987 100644 --- a/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py +++ b/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py @@ -1,3 +1,21 @@ +# /// script +# requires-python = ">=3.10" +# dependencies = [ +# "torch", +# "torchvision", +# "diffusers", +# "transformers", +# "accelerate", +# "peft", +# "safetensors", +# "huggingface_hub", +# "datasets", +# "Pillow", +# "tqdm", +# "bitsandbytes" +# ] +# /// + #!/usr/bin/env python # coding=utf-8 # Copyright 2025 The HuggingFace Inc. team. All rights reserved. From 55ab58d4c03aaeecdbb3c26c78116e079873cf36 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?apolin=C3=A1rio?= Date: Thu, 31 Jul 2025 17:27:11 +0200 Subject: [PATCH 2/4] Update train_dreambooth_lora_flux_advanced.py --- .../train_dreambooth_lora_flux_advanced.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py b/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py index 4150c66c3987..c17f2f9e72b3 100644 --- a/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py +++ b/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py @@ -3,7 +3,7 @@ # dependencies = [ # "torch", # "torchvision", -# "diffusers", +# "git+https://github.com/huggingface/diffusers.git", # "transformers", # "accelerate", # "peft", From 1ba69ec310e7cc8f0af2cb06a7a96a8cbfa9645a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?apolin=C3=A1rio?= Date: Thu, 31 Jul 2025 17:28:40 +0200 Subject: [PATCH 3/4] Update train_dreambooth_lora_flux_advanced.py --- .../train_dreambooth_lora_flux_advanced.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py b/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py index c17f2f9e72b3..9a7658c76641 100644 --- a/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py +++ b/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py @@ -3,7 +3,7 @@ # dependencies = [ # "torch", # "torchvision", -# "git+https://github.com/huggingface/diffusers.git", +# "diffusers @ git+https://github.com/huggingface/diffusers.git@main", # "transformers", # "accelerate", # "peft", From 1afd6c1721323a660bca1551cc6117ce2469b0c5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?apolin=C3=A1rio?= Date: Thu, 31 Jul 2025 17:41:52 +0200 Subject: [PATCH 4/4] Update train_dreambooth_lora_flux_advanced.py --- .../train_dreambooth_lora_flux_advanced.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py b/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py index 9a7658c76641..2b3bdb563f3b 100644 --- a/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py +++ b/examples/advanced_diffusion_training/train_dreambooth_lora_flux_advanced.py @@ -12,7 +12,10 @@ # "datasets", # "Pillow", # "tqdm", -# "bitsandbytes" +# "bitsandbytes", +# "sentencepiece", +# "protobuf", +# "prodigyopt", # ] # ///