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 c17f2f9e72..9a7658c766 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",