21 Commits

Author SHA1 Message Date
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