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}.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)
|
||||
|
||||
|
||||
|
|
|
@ -257,7 +257,7 @@ def controllora_sdxl_config(
|
|||
config_name: ControlLoraResolvedConfig(
|
||||
scale=config.scale,
|
||||
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()
|
||||
}
|
||||
|
@ -1155,7 +1155,7 @@ def test_diffusion_controlnet_stack(
|
|||
|
||||
|
||||
@no_grad()
|
||||
def test_diffusion_sdxl_controllora(
|
||||
def test_diffusion_sdxl_control_lora(
|
||||
controllora_sdxl_config: tuple[Image.Image, dict[str, ControlLoraResolvedConfig]],
|
||||
sdxl_ddim_lda_fp16_fix: StableDiffusion_XL,
|
||||
) -> None:
|
||||
|
|
Loading…
Reference in a new issue