Jaret Burkett 7195abc32b Performance fixes and bug fixes

2026-08-28 22:21:09 -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_pixart2.py

Save entire pixart model again

2024-07-07 07:56:48 -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

shrink_pixart_sm.py

Varous bug fixes

2024-07-10 15:20:04 -06:00

shrink_pixart.py

Save entire pixart model again

2024-07-07 07:56:48 -06:00

stage_profiler.py

Fix quantization issues

2026-08-28 10:48:34 -06:00

test_bucket_dataloader.py

Adjust dataloader tester to handle videos to test them

2025-10-21 14:47:23 -06:00

test_lora_compile_scalars.py

Restore device runtime scales for compiled adapters (#963)

2026-07-18 08:19:57 -06:00

test_ltx_dataloader.py

Add LTX-2 Support (#644)

2026-01-13 04:55:30 -07: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_model_loading.py

Performance fixes and bug fixes

2026-08-28 22:21:09 -06:00

test_vae_cycle.py

Handle conversions back to ldm for saving

2023-07-18 19:34:35 -06:00

test_vae.py

Added cogview4. Loss still needs work.

2025-03-04 18:43:52 -07:00