Compare commits

...

5 Commits

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

View File

@@ -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.