ai-toolkit/testing
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
..
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 Fixed some mismatched weights by adjusting tolerance. The mismatch ironically made the models better lol 2023-08-29 15:20:03 -06:00
test_bucket_dataloader.py 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
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