Ryan Dick e2f05d0800 Add unit tests for LoKR patch layers. The new tests trigger a bug when LoKR layers are applied to BnB-quantized layers (also impacts several other LoRA variant types).

2025-01-22 09:20:40 +11:00

..

data

Get convert_flux_kohya_state_dict_to_invoke_format(...) working, with unit tests.

2024-09-15 04:39:56 +03:00

load/model_cache

Add unit tests for LoKR patch layers. The new tests trigger a bug when LoKR layers are applied to BnB-quantized layers (also impacts several other LoRA variant types).

2025-01-22 09:20:40 +11:00

model_loading

tidy(mm): remove convenience methods from high level model manager service

2024-03-07 10:56:59 +11:00

model_metadata

Add unit test

2024-03-19 01:16:11 -04:00

util

Update HF download logic to work for black-forest-labs/FLUX.1-schnell.

2024-08-26 20:17:50 -04:00

model_manager_fixtures.py

Add keep_ram_copy_of_weights config option.

2025-01-16 15:35:25 +00:00

test_libc_util.py

Tidy names and locations of modules

2024-03-01 10:42:33 +11:00

test_memory_snapshot.py

final tidying before marking PR as ready for review

2024-03-01 10:42:33 +11:00

test_model_load_optimization.py

Tidy names and locations of modules

2024-03-01 10:42:33 +11:00