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

check hash of downloaded LoRA weights, update DPO refs

Browse source 
(the DPO LoRA weights have changed: 2699b36e22)
This commit is contained in:
Pierre Chapuis 2024-02-23 11:23:20 +01:00
parent 28f9368c93
commit 8f614e7647
3 changed files with 24 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
scripts/prepare_test_weights.py
View file

@ -232,17 +232,25 @@ def download_vae_ft_mse():
def download_loras(): def download_loras():
dest_folder = os.path.join(test_weights_dir, "loras", "pokemon-lora") dest_folder = os.path.join(test_weights_dir, "loras", "pokemon-lora")
download_file("https://huggingface.co/pcuenq/pokemon-lora/resolve/main/pytorch_lora_weights.bin", dest_folder) download_file(
"https://huggingface.co/pcuenq/pokemon-lora/resolve/main/pytorch_lora_weights.bin",
dest_folder,
expected_hash="89992ea6",
)
dest_folder = os.path.join(test_weights_dir, "loras", "dpo-lora") dest_folder = os.path.join(test_weights_dir, "loras", "dpo-lora")
download_file( download_file(
"https://huggingface.co/radames/sdxl-DPO-LoRA/resolve/main/pytorch_lora_weights.safetensors", dest_folder "https://huggingface.co/radames/sdxl-DPO-LoRA/resolve/main/pytorch_lora_weights.safetensors",
dest_folder,
expected_hash="a51e9144",
) )
dest_folder = os.path.join(test_weights_dir, "loras", "sliders") dest_folder = os.path.join(test_weights_dir, "loras", "sliders")
download_file("https://sliders.baulab.info/weights/xl_sliders/age.pt", dest_folder) download_file("https://sliders.baulab.info/weights/xl_sliders/age.pt", dest_folder, expected_hash="908f07d3")
download_file("https://sliders.baulab.info/weights/xl_sliders/cartoon_style.pt", dest_folder) download_file(
download_file("https://sliders.baulab.info/weights/xl_sliders/eyesize.pt", dest_folder) "https://sliders.baulab.info/weights/xl_sliders/cartoon_style.pt", dest_folder, expected_hash="25652004"
)
download_file("https://sliders.baulab.info/weights/xl_sliders/eyesize.pt", dest_folder, expected_hash="ee170e4d")
def download_preprocessors(): def download_preprocessors():
@ -276,15 +284,17 @@ def download_controlnet():
def download_control_lora_fooocus(): def download_control_lora_fooocus():
base_folder = os.path.join(test_weights_dir, "lllyasviel", "misc") base_folder = os.path.join(test_weights_dir, "lllyasviel", "misc")
control_loras = [
"control-lora-canny-rank128.safetensors",
"fooocus_xl_cpds_128.safetensors",
]
for control_lora in control_loras:
download_file( download_file(
url=f"https://huggingface.co/lllyasviel/misc/resolve/main/{control_lora}", url=f"https://huggingface.co/lllyasviel/misc/resolve/main/control-lora-canny-rank128.safetensors",
dest_folder=base_folder, dest_folder=base_folder,
expected_hash="4d505134",
)
download_file(
url=f"https://huggingface.co/lllyasviel/misc/resolve/main/fooocus_xl_cpds_128.safetensors",
dest_folder=base_folder,
expected_hash="d81aa461",
) )

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.6 MiB

After

Width:  |  Height:  |  Size: 1.5 MiB

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.7 MiB