Jaret Burkett
|
01cf480233
|
Add FLUX.2 official weights
|
2025-11-25 08:52:19 -07:00 |
|
Jaret Burkett
|
dadbeda197
|
Update test weights
|
2025-11-23 10:51:50 -07:00 |
|
Jaret Burkett
|
af8e9ea149
|
Add initial support for FLUX.2
|
2025-11-18 11:17:38 -07:00 |
|
Jaret Burkett
|
2e7b2d9926
|
Added Differential Guidance training target
|
2025-11-10 09:38:25 -07:00 |
|
Jaret Burkett
|
6f308fc46e
|
When soing guidance loss, make CFG zero an optional target instead of a forced one.
|
2025-11-04 09:16:15 -07:00 |
|
Jaret Burkett
|
c984369294
|
Fixed resizing of control image resolution for Qwen Image Edit 2509 when using match_target_res
|
2025-10-30 06:30:01 -06:00 |
|
Jaret Burkett
|
42e5e3cd1c
|
Adjust DFE to handle 5 dimension latent spaces
|
2025-10-27 07:48:44 -06:00 |
|
Jaret Burkett
|
80418209b8
|
Fixe a variable that could nt be declared when doing blank prompt preservation
|
2025-10-23 16:19:56 -06:00 |
|
Jaret Burkett
|
ee206cfa18
|
Added blank prompt preservation
|
2025-10-22 14:55:13 -06:00 |
|
Jaret Burkett
|
ff14cd6343
|
Fix check for making sure vae is on the right device.
|
2025-10-21 14:49:20 -06:00 |
|
Jaret Burkett
|
76ce757e0c
|
Added initial support for layer offloading wit Wan 2.2 14B models.
|
2025-10-20 14:54:30 -06:00 |
|
Jaret Burkett
|
b7f85928f3
|
Fix issue with chroma when not quantizing
|
2025-10-19 12:13:05 -06:00 |
|
Jaret Burkett
|
91b87e06a1
|
Reordered logs
|
2025-10-15 09:15:06 -06:00 |
|
Jaret Burkett
|
7e34a03113
|
Added queing system to the UI
|
2025-10-14 12:00:42 -06:00 |
|
Jaret Burkett
|
0c9e1c3deb
|
Fixed some fringe cases for qwen image edit.
|
2025-10-13 17:10:46 +00:00 |
|
Jaret Burkett
|
e9c4d94256
|
Allow for matching target resolution with control images for Qwen Image Edit 2509
|
2025-10-10 14:24:27 -06:00 |
|
Jaret Burkett
|
1bc6dee127
|
Change auto_memory to be layer_offloading and allow you to set the amount to unload
|
2025-10-10 13:12:32 -06:00 |
|
Jaret Burkett
|
8068755b0a
|
Fixed issue with wan 2.2 getting stuck on CPU
|
2025-10-09 17:24:25 +00:00 |
|
Jaret Burkett
|
4e5707854f
|
Initial support for RamTorch. Still a WIP
|
2025-10-05 13:03:26 -06:00 |
|
Jaret Burkett
|
b7c04efb44
|
A commit with the adits properly named improvements to qwen image edit plus workflow. Fixed a bug. Dont norm the cfg
|
2025-10-01 14:13:15 -06:00 |
|
Jaret Burkett
|
3086a58e5b
|
git status
|
2025-10-01 14:12:17 -06:00 |
|
Jaret Burkett
|
2ba4000704
|
Allow masked losses with video models
|
2025-09-30 14:57:07 -06:00 |
|
Jaret Burkett
|
67ed563e03
|
fix issue with multi batch size on qwen-image-edit-plus
|
2025-09-30 09:04:56 -06:00 |
|
Jaret Burkett
|
6da417261c
|
Add extra detachments just to be sure on qiep
|
2025-09-27 08:53:59 -06:00 |
|
Jaret Burkett
|
e04f55c553
|
Fixed scaling issue with control images
|
2025-09-26 11:49:53 -06:00 |
|
Jaret Burkett
|
454be0958a
|
Initial support for qwen image edit plus
|
2025-09-24 11:39:10 -06:00 |
|
Jaret Burkett
|
f74475161e
|
Add stepped loss type
|
2025-09-22 15:50:12 -06:00 |
|
Jaret Burkett
|
28728a1e92
|
Added experimental dfe 5
|
2025-09-21 10:48:52 -06:00 |
|
Jaret Burkett
|
20dfe1b4d5
|
Small double tap of detach on qwen just for good measure
|
2025-09-18 08:22:04 -06:00 |
|
Jaret Burkett
|
3cdf50cbfc
|
Merge pull request #426 from squewel/prior_reg
Dataset-level prior regularization
|
2025-09-18 03:03:18 -06:00 |
max
|
e4ae97e790
|
add dataset-level distillation-style regularization
|
2025-09-18 01:11:19 +03:00 |
|
Jaret Burkett
|
24a576ad07
|
Regularize the slider targets.
|
2025-09-17 09:36:33 -06:00 |
|
Jaret Burkett
|
218f673e3d
|
Added support for new concept slider training script to CLI and UI
|
2025-09-16 10:22:34 -06:00 |
|
Jaret Burkett
|
b95c17dc17
|
Add initial support for chroma radiance
|
2025-09-10 08:41:05 -06:00 |
|
Jaret Burkett
|
645046701b
|
Comment out fast stop watcher. Could potentiallty be causing some weird issues. Need to investigate.
|
2025-09-04 08:26:57 -06:00 |
|
Jaret Burkett
|
193c1b2dfa
|
Add a watcher to constantly check for stop signal from the UI. This will force a stop within 2 seconds instead of having to wait on a long hung process.
|
2025-08-31 16:58:01 -06:00 |
|
Jaret Burkett
|
056711d4ed
|
Fix issue with wan22 14b that woudl load both transformers temporarily resulting in oom on 24GB.
|
2025-08-28 13:06:31 -06:00 |
|
Jaret Burkett
|
1f541bc5d8
|
Changes to handle a different DFE arch
|
2025-08-27 11:05:16 -06:00 |
|
Jaret Burkett
|
119653c3f2
|
Force width, height, and num frames to always be the proper sizes for Wan 2.2 models
|
2025-08-25 10:33:28 -06:00 |
|
Jaret Burkett
|
e1fd411665
|
Added support for Chroma1 official release. Will still use single file verstion instead of the diffusers version.
|
2025-08-23 09:06:28 -06:00 |
|
Jaret Burkett
|
aa99784b89
|
Add control to prompot encodings in the trainer when not cached
|
2025-08-21 16:52:13 -06:00 |
|
Jaret Burkett
|
bf2700f7be
|
Initial support for finetuning qwen image. Will only work with caching for now, need to add controls everywhere.
|
2025-08-21 16:41:17 -06:00 |
|
Jaret Burkett
|
83deaec417
|
Minor bug fixes
|
2025-08-21 08:05:34 -06:00 |
|
Jaret Burkett
|
d2bbe1872c
|
Add support for fine tuning Wan 2.2 I2V 14B
|
2025-08-18 11:43:32 -06:00 |
|
Jaret Burkett
|
b3e666daf4
|
Fix issue with wan22 14b where timesteps were generated not in the current boundary.
|
2025-08-16 21:16:48 -06:00 |
|
Jaret Burkett
|
6fffadfc0e
|
Fixed a bug that prevented training just one stage of Wan 2.2 14b
|
2025-08-16 18:07:21 -06:00 |
|
Jaret Burkett
|
8ea2cf00f6
|
Added training to the ui. Still testing, but everything seems to be working.
|
2025-08-16 05:51:37 -06:00 |
|
Jaret Burkett
|
1c96b95617
|
Fix issue where sometimes the transformer does not get loaded properly.
|
2025-08-14 14:24:41 -06:00 |
|
Jaret Burkett
|
3413fa537f
|
Wan22 14b training is working, still need tons of testing and some bug fixes
|
2025-08-14 13:03:27 -06:00 |
|
Jaret Burkett
|
259d68d440
|
Added a flushg during sampling to prevent spikes on low vram qwen
|
2025-08-12 12:57:18 -06:00 |