ai-toolkit/testing
Jaret Burkett 3f3636b788 Bug fixes and little improvements here and there. 2024-06-08 06:24:20 -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 Bug fixes and minor features 2024-04-25 06:14:31 -06:00
test_bucket_dataloader.py Bug fixes and little improvements here and there. 2024-06-08 06:24:20 -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_cycle.py Handle conversions back to ldm for saving 2023-07-18 19:34:35 -06:00