Skip to content

Commit 50cf285

Browse files
Kevin Turnerpsychedelicious
authored andcommitted
fix: group aitoolkit lora layers
1 parent a214f4f commit 50cf285

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/backend/patches/lora_conversions/test_flux_aitoolkit_lora_conversion_utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,9 +48,9 @@ def test_flux_aitoolkit_transformer_state_dict_is_in_invoke_format():
4848
model_keys = set(model.state_dict().keys())
4949

5050
for converted_key_prefix in converted_key_prefixes:
51-
assert any(
52-
model_key.startswith(converted_key_prefix) for model_key in model_keys
53-
), f"'{converted_key_prefix}' did not match any model keys."
51+
assert any(model_key.startswith(converted_key_prefix) for model_key in model_keys), (
52+
f"'{converted_key_prefix}' did not match any model keys."
53+
)
5454

5555

5656
def test_lora_model_from_flux_aitoolkit_state_dict():

0 commit comments

Comments
 (0)