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 a30624e35a8d..f01a08d8b219 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,24 @@ +# /// script +# requires-python = ">=3.10" +# dependencies = [ +# "torch", +# "torchvision", +# "diffusers @ git+https://github.com/huggingface/diffusers.git@main", +# "transformers", +# "accelerate", +# "peft", +# "safetensors", +# "huggingface_hub", +# "datasets", +# "Pillow", +# "tqdm", +# "bitsandbytes", +# "sentencepiece", +# "protobuf", +# "prodigyopt", +# ] +# /// + #!/usr/bin/env python # coding=utf-8 # Copyright 2025 The HuggingFace Inc. team. All rights reserved.