1a1f97263d
CHORE Replace deprecated torch_dtype with dtype ( #2837 )
...
Note: Diffusers is left as is for now, might need an update later.
2025-10-16 14:59:09 +02:00
68265a1583
ENH XPU support for training dreambooth ( #2696 )
...
---------
Signed-off-by: Yao, Matrix <matrix.yao@intel.com >
2025-08-04 11:42:45 +02:00
9f0cfc9919
Don't use deprecated Repository
anymore ( #1641 )
...
* Don't use deprecated Repository anymore
* oops
2024-04-12 13:51:37 +02:00
a18734d87a
Update style with ruff 0.2.2 ( #1565 )
...
This is necessary to add to main fast, or else all branches from main
will require these changes to pass the quality checks.
2024-03-15 10:20:41 +01:00
fc78a2491e
MNT Move code quality fully to ruff ( #1421 )
2024-02-07 12:52:35 +01:00
98429b8184
Fix: TorchTracemalloc ruins Windows performance ( #1126 )
...
* feat: added tracemalloc arg to train_dreambooth
* fix: added help for arg
* fix: changed arg name
* fix formatting
* fix: import order
2023-11-14 17:04:32 +05:30
49ddefa834
Add num_dataloader_workers arg to dreambooth script ( #1107 )
...
This is especially important for Windows users, who may have to set the
number of workers to 0.
2023-11-10 14:21:14 +01:00
b5e341bb8a
Added wandb support for lora train_dreambooth ( #639 )
...
* Update train_dreambooth.py
Accelerator init updated from logging_dir to project_dir. Newer versions of accelerate uses project_dir. logging_dir is deprecated
* Bugfix: Adapter name variable inserted, when changing LORA_ADAPTER_NAME it causes error
* Adapter name added as kwarg
* Black code formatted
* Style & Quality check
* Wandb import added for logging and project initialization
* Wandb import added for logging and project initialization
* fix project_name
* print tqdm progress to wandb
2023-07-11 13:56:03 +05:30
18544647ac
Update train_dreambooth.py ( #624 )
...
Accelerator init updated from logging_dir to project_dir. Newer versions of accelerate uses project_dir. logging_dir is deprecated
2023-06-26 18:18:24 +05:30
70af02a2bc
fix and update examples and readme ( #295 )
...
* fix and update examples and readme
* fix formatting issues
* add weighted lora adapters support
* Update README.md
2023-04-13 00:17:13 +05:30
891584c8d9
fix ci dreambooth
2023-03-28 13:55:43 +00:00
b5b3ae3cbe
Update train_dreambooth.py
2023-03-21 21:38:15 +08:00
b9451ab458
fixing issues and quality ✨
2023-03-07 14:04:19 +05:30
80e1b262e5
fix validation_steps handling in dreambooth example
2023-01-31 10:59:46 +08:00
a17cea104e
add validation images for lora training
2023-01-30 17:18:43 +08:00
fd8d3804c7
Update train_dreambooth.py
2023-01-15 15:50:25 +01:00
086b329c1a
addressing comments and renaming pet
to peft
2023-01-15 14:39:56 +01:00
4d76bbac14
Update train_dreambooth.py
2023-01-13 00:26:58 +01:00
c8d3def60e
minor fixes
2023-01-04 15:16:32 +05:30
82ef390fb9
✨
2023-01-03 22:18:35 +05:30
71d6c074de
add code and update README
2023-01-03 22:01:41 +05:30