ai-toolkit/testing
petmycat 02c06cb996
feature: TST v4 (#4)
Co-authored-by: James <1561862923@qq.com>
2026-08-14 19:35:13 +08: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 Adjust dataloader tester to handle videos to test them 2025-10-21 14:47:23 -06:00
test_ideogram4_trigger_activator.py feature: TST v4 (#4) 2026-08-14 19:35:13 +08: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_sd_trainer_three_phase_runtime.py feature: TST v4 (#4) 2026-08-14 19:35:13 +08:00
test_three_phase_trigger_training.py feature: TST v4 (#4) 2026-08-14 19:35:13 +08:00
test_trigger_binding.py feature: TST v4 (#4) 2026-08-14 19:35:13 +08:00
test_trigger_binding_artifacts.py feature: TST v4 (#4) 2026-08-14 19:35:13 +08:00
test_trigger_binding_losses.py feature: TST v4 (#4) 2026-08-14 19:35:13 +08:00
test_trigger_selective_training.py Feat/experimental (#3) 2026-08-14 10:57:48 +08: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