mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
fix conversion_script bug + rename control_lora e2e test
This commit is contained in:
parent
7fe392298a
commit
00270604ef
|
@ -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)
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -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:
|
||||||
|
|
Loading…
Reference in a new issue