Jaret Burkett b829983b16 Added ability to load video datasets and train with them		2025-03-19 09:54:26 -06:00
..
compare_keys.py	Added method to get specific keys from model	2023-08-15 14:51:04 -06:00
generate_lora_mapping.py	Corrected key saving and loading to better match kohya	2023-09-04 00:22:34 -06:00
generate_weight_mappings.py	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
merge_in_text_encoder_adapter.py	Added caching to image sizes so we dont do it every time.	2024-07-15 19:07:41 -06:00
shrink_pixart.py	Save entire pixart model again	2024-07-07 07:56:48 -06:00
shrink_pixart2.py	Save entire pixart model again	2024-07-07 07:56:48 -06:00
shrink_pixart_sm.py	Varous bug fixes	2024-07-10 15:20:04 -06:00
shrink_pixart_sm2.py	Varous bug fixes	2024-07-10 15:20:04 -06:00
shrink_pixart_sm3.py	Varous bug fixes	2024-07-10 15:20:04 -06:00
test_bucket_dataloader.py	Added ability to load video datasets and train with them	2025-03-19 09:54:26 -06:00
test_model_load_save.py	Fixed some mismatched weights by adjusting tolerance. The mismatch ironically made the models better lol	2023-08-29 15:20:03 -06:00
test_vae.py	Added cogview4. Loss still needs work.	2025-03-04 18:43:52 -07:00
test_vae_cycle.py	Handle conversions back to ldm for saving	2023-07-18 19:34:35 -06:00