Merge branch 'main' into add-uv-script-test

Update train_dreambooth_lora_flux_advanced.py
2025-07-31 22:10:11 +05:30 · 2025-07-31 17:41:52 +02:00 · 2025-07-31 17:28:40 +02:00 · 2025-07-31 17:27:11 +02:00 · 2025-07-31 17:17:02 +02:00
1 changed files with 21 additions and 0 deletions
@@ -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.
Author	SHA1	Message	Date
Sayak Paul	82f0b2a11d	Merge branch 'main' into add-uv-script-test	2025-07-31 22:10:11 +05:30
apolinário	1afd6c1721	Update train_dreambooth_lora_flux_advanced.py	2025-07-31 17:41:52 +02:00
apolinário	1ba69ec310	Update train_dreambooth_lora_flux_advanced.py	2025-07-31 17:28:40 +02:00
apolinário	55ab58d4c0	Update train_dreambooth_lora_flux_advanced.py	2025-07-31 17:27:11 +02:00
apolinário	67a7358ee8	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	2025-07-31 17:17:02 +02:00