Jaret Burkett 39870411d8 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

..

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

test_bucket_dataloader.py

Numerous fixes for time sampling. Still not perfect

2023-11-28 07:34:43 -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_vae_cycle.py

Handle conversions back to ldm for saving

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