Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-09-19 19:05:28 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

fix slider loras test

Browse source
This commit is contained in:
Pierre Chapuis 2024-03-05 16:52:21 +01:00
parent 8c7fcbc00f
commit 72fa17df48
2 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/e2e/test_diffusion.py
View file

@ -1430,14 +1430,18 @@ def test_diffusion_sdxl_multiple_loras(
) -> None: ) -> None:
sdxl = sdxl_ddim sdxl = sdxl_ddim
expected_image = expected_sdxl_multi_loras expected_image = expected_sdxl_multi_loras
_, dpo = lora_data_dpo _, dpo_weights = lora_data_dpo
loras, scales = lora_sliders slider_loras, slider_scales = lora_sliders
loras["dpo"] = dpo
manager = SDLoraManager(sdxl) manager = SDLoraManager(sdxl)
for lora_name, lora_weights in loras.items(): for lora_name, lora_weights in slider_loras.items():
manager.add_loras(lora_name, lora_weights, scales[lora_name]) manager.add_loras(
manager.add_loras("dpo", dpo, 1.4) lora_name,
lora_weights,
slider_scales[lora_name],
unet_inclusions=["SelfAttention", "ResidualBlock", "Downsample", "Upsample"],
)
manager.add_loras("dpo", dpo_weights, 1.4, unet_inclusions=["CrossAttentionBlock"])
# parameters are the same as https://huggingface.co/radames/sdxl-DPO-LoRA # parameters are the same as https://huggingface.co/radames/sdxl-DPO-LoRA
# except that we are using DDIM instead of sde-dpmsolver++ # except that we are using DDIM instead of sde-dpmsolver++

BIN
tests/e2e/test_diffusion_ref/expected_sdxl_multi_loras.png
View file

Binary file not shown.
Side by side Swipe Overlay

Before

Width:  |  Height:  |  Size: 1.7 MiB

After

Width:  |  Height:  |  Size: 1.8 MiB