Jaret Burkett
5b5aadadb8
Add LTX-2 Support ( #644 )
...
* WIP, adding support for LTX2
* Training on images working
* Fix loading comfy models
* Handle converting and deconverting lora so it matches original format
* Reworked ui to habdle ltx and propert dataset default overwriting.
* Update the way lokr saves to it is more compatable with comfy
* Audio loading and synchronization/resampling is working
* Add audio to training. Does it work? Maybe, still testing.
* Fixed fps default issue for sound
* Have ui set fps for accurate audio mapping on ltx
* Added audio procession options to the ui for ltx
* Clean up requirements
2026-01-13 04:55:30 -07:00
Jaret Burkett
5123090f6c
Adjust dataloader tester to handle videos to test them
2025-10-21 14:47:23 -06:00
Jaret Burkett
bfe29e2151
Removed all submodules. Submodule free now, yay.
2025-04-18 10:39:15 -06:00
Jaret Burkett
b829983b16
Added ability to load video datasets and train with them
2025-03-19 09:54:26 -06:00
Jaret Burkett
6f6fb90812
Added cogview4. Loss still needs work.
2025-03-04 18:43:52 -07:00
Jaret Burkett
338c77d677
Fixed breaking change with diffusers. Allow flowmatch on normal stable diffusion models.
2024-08-22 14:36:22 -06:00
Jaret Burkett
03613c523f
Bugfixes and cleanup
2024-08-01 11:45:12 -06:00
Jaret Burkett
58dffd43a8
Added caching to image sizes so we dont do it every time.
2024-07-15 19:07:41 -06:00
Jaret Burkett
c062b7716c
Varous bug fixes
2024-07-10 15:20:04 -06:00
Jaret Burkett
045e4a6e15
Save entire pixart model again
2024-07-07 07:56:48 -06:00
Jaret Burkett
cab8a1c7b8
WIP to add the caption_proj weight to pixart sigma TE adapter
2024-07-06 13:00:21 -06:00
Jaret Burkett
acb06d6ff3
Bug fixes
2024-07-03 10:56:34 -06:00
Jaret Burkett
64f2b085b7
Minor fixes
2024-06-23 14:47:40 -06:00
Jaret Burkett
3f3636b788
Bug fixes and little improvements here and there.
2024-06-08 06:24:20 -06:00
Jaret Burkett
833c833f28
WIP on SAFE encoder. Work on fp16 training improvements. Various other tweaks and improvements
2024-05-27 10:50:24 -06:00
Jaret Burkett
5a45c709cd
Work on ipadapters and custom adapters
2024-05-13 06:37:54 -06:00
Jaret Burkett
5da3613e0b
Bug fixes and minor features
2024-04-25 06:14:31 -06:00
Jaret Burkett
016687bda1
Adapter work. Bug fixes. Auto adjust LR when resuming optimizer.
2024-03-17 10:21:47 -06:00
Jaret Burkett
39870411d8
More guidance work. Improved LoRA module resolver for unet. Added vega mappings and LoRA training for it. Various other bigfixes and changes
2023-12-15 06:02:10 -07:00
Jaret Burkett
792a5e37e2
Numerous fixes for time sampling. Still not perfect
2023-11-28 07:34:43 -07:00
Jaret Burkett
a8b3b8b8da
Fixed weight mapping for refiner
2023-11-06 07:37:47 -07:00
Jaret Burkett
93ea955d7c
Added refiner fine tuning. Works, but needs some polish.
2023-11-05 17:15:03 -07:00
Jaret Burkett
d35733ac06
Added support for training ssd-1B. Added support for saving models into diffusers format. We can currently save in safetensors format for ssd-1b, but diffusers cannot load it yet.
2023-11-03 05:01:16 -06:00
Jaret Burkett
579650eaf8
Fixed big issue with bucketing dataloader and added random cripping to a point of interest
2023-10-02 18:31:08 -06:00
Jaret Burkett
19255cdc7c
Bugfixes. Added small augmentations to dataloader. Will switch to abluminations soon though. Added ability to adjust step count on start to override what is in the file
2023-09-20 05:30:10 -06:00
Jaret Burkett
fa8fc32c0a
Corrected key saving and loading to better match kohya
2023-09-04 00:22:34 -06:00
Jaret Burkett
4ca819a05e
Fixes for dataloader
2023-08-31 04:54:10 -06:00
Jaret Burkett
33267e117c
Reworked bucket loader to scale buckets to pixels amounts not just minimum size. Makes the network more consistant
2023-08-30 14:52:12 -06:00
Jaret Burkett
836fee47a6
Fixed some mismatched weights by adjusting tolerance. The mismatch ironically made the models better lol
2023-08-29 15:20:03 -06:00
Jaret Burkett
14ff51ceb4
fixed issues with converting and saving models. Cleaned keys. Improved testing for cycle load saving.
2023-08-29 12:31:19 -06:00
Jaret Burkett
714854ee86
Hude rework to move the batch to a DTO to make it far more modular to the future ui
2023-08-29 10:22:19 -06:00
Jaret Burkett
bee0b6a235
Added converters for all stable diffusion models to convert back to ldm format from diffusers.
2023-08-28 16:12:32 -06:00
Jaret Burkett
fab7c2b04a
Fixed issue with key mapping from diffusers back to ldm
2023-08-28 14:01:26 -06:00
Jaret Burkett
8105c05c12
Added bucketting capabilities to dataloader. Finally have full planned capability. noice
2023-08-26 16:36:32 -06:00
Jaret Burkett
36ba08d3fa
Added a converter back to ldm from diffusers for sdxl. Can finally get to training it properly
2023-08-21 16:22:01 -06:00
Jaret Burkett
55a5fcc7d9
Added method to get specific keys from model
2023-08-15 14:51:04 -06:00
Jaret Burkett
c1b1e800df
Updated module names to be compatable with ney koyha unet
2023-07-31 13:16:29 -06:00
Jaret Burkett
8d8229dfc0
Handle conversions back to ldm for saving
2023-07-18 19:34:35 -06:00
