Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-21 21:58:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix conversion_script bug + rename control_lora e2e test

Browse source
This commit is contained in:
Laurent 2024-02-14 16:32:34 +00:00 committed by Laureηt
parent 7fe392298a
commit 00270604ef
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
scripts/conversion/convert_fooocus_control_lora.py
View file

@ -211,7 +211,7 @@ def convert_lora_layers(
f"{path}.down": lora_layer.down.weight, f"{path}.down": lora_layer.down.weight,
f"{path}.up": lora_layer.up.weight, f"{path}.up": lora_layer.up.weight,
} }
state_dict = {simplify_key(key, "ControlLora"): param for key, param in state_dict.items()} state_dict = {simplify_key(key, "ControlLora."): param for key, param in state_dict.items()}
refiners_state_dict.update(state_dict) refiners_state_dict.update(state_dict)

4
tests/e2e/test_diffusion.py
View file

@ -257,7 +257,7 @@ def controllora_sdxl_config(
config_name: ControlLoraResolvedConfig( config_name: ControlLoraResolvedConfig(
scale=config.scale, scale=config.scale,
condition_image=Image.open(ref_path / config.condition_path).convert("RGB"), condition_image=Image.open(ref_path / config.condition_path).convert("RGB"),
weights_path=test_weights_path / "control_lora" / config.weights_path, weights_path=test_weights_path / "control-loras" / config.weights_path,
) )
for config_name, config in configs.items() for config_name, config in configs.items()
} }
@ -1155,7 +1155,7 @@ def test_diffusion_controlnet_stack(
@no_grad() @no_grad()
def test_diffusion_sdxl_controllora( def test_diffusion_sdxl_control_lora(
controllora_sdxl_config: tuple[Image.Image, dict[str, ControlLoraResolvedConfig]], controllora_sdxl_config: tuple[Image.Image, dict[str, ControlLoraResolvedConfig]],
sdxl_ddim_lda_fp16_fix: StableDiffusion_XL, sdxl_ddim_lda_fp16_fix: StableDiffusion_XL,
) -> None: ) -> None: