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

fix DPO LoRA loading in tests

Browse source
This commit is contained in:
Pierre Chapuis 2024-03-05 15:14:34 +01:00
parent ed8ec26e63
commit c383ff6cf4
2 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/e2e/test_diffusion.py
View file

@ -1395,7 +1395,8 @@ def test_diffusion_sdxl_lora(
prompt = "professional portrait photo of a girl, photograph, highly detailed face, depth of field, moody light, golden hour, style by Dan Winters, Russell James, Steve McCurry, centered, extremely detailed, Nikon D850, award winning photography"
negative_prompt = "3d render, cartoon, drawing, art, low light, blur, pixelated, low resolution, black and white"
SDLoraManager(sdxl).add_loras("dpo", lora_weights, scale=lora_scale)
manager = SDLoraManager(sdxl, unet_inclusions=["CrossAttentionBlock"])
manager.add_loras("dpo", lora_weights, scale=lora_scale)
clip_text_embedding, pooled_text_embedding = sdxl.compute_clip_text_embedding(
text=prompt, negative_text=negative_prompt

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

Binary file not shown.
Side by side Swipe Overlay

Before

Width:  |  Height:  |  Size: 1.5 MiB

After

Width:  |  Height:  |  Size: 1.6 MiB