Skip to content

Commit dc53dfe

Browse files
committed
Add kohya embedders conversion test
1 parent d24db27 commit dc53dfe

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/lora/test_lora_layers_flux.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -907,6 +907,14 @@ def test_flux_kohya_with_text_encoder(self):
907907

908908
assert max_diff < 1e-3
909909

910+
def test_flux_kohya_embedders_conversion(self):
911+
"""Test that embedders load without throwing errors"""
912+
self.pipeline.load_lora_weights("rockerBOO/flux-bpo-po-lora")
913+
self.pipeline.unload_lora_weights()
914+
915+
assert True
916+
917+
910918
def test_flux_xlabs(self):
911919
self.pipeline.load_lora_weights("XLabs-AI/flux-lora-collection", weight_name="disney_lora.safetensors")
912920
self.pipeline.fuse_lora()

0 commit comments

Comments
 (0)