diff --git a/scripts/prepare_test_weights.py b/scripts/prepare_test_weights.py index 77783eb..6d41982 100644 --- a/scripts/prepare_test_weights.py +++ b/scripts/prepare_test_weights.py @@ -232,17 +232,25 @@ def download_vae_ft_mse(): def download_loras(): 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") 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") - download_file("https://sliders.baulab.info/weights/xl_sliders/age.pt", dest_folder) - download_file("https://sliders.baulab.info/weights/xl_sliders/cartoon_style.pt", dest_folder) - download_file("https://sliders.baulab.info/weights/xl_sliders/eyesize.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, expected_hash="25652004" + ) + download_file("https://sliders.baulab.info/weights/xl_sliders/eyesize.pt", dest_folder, expected_hash="ee170e4d") def download_preprocessors(): @@ -276,16 +284,18 @@ def download_controlnet(): def download_control_lora_fooocus(): 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( - url=f"https://huggingface.co/lllyasviel/misc/resolve/main/{control_lora}", - dest_folder=base_folder, - ) + download_file( + url=f"https://huggingface.co/lllyasviel/misc/resolve/main/control-lora-canny-rank128.safetensors", + 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", + ) def download_unclip(): diff --git a/tests/e2e/test_diffusion_ref/expected_sdxl_dpo_lora.png b/tests/e2e/test_diffusion_ref/expected_sdxl_dpo_lora.png index 07d4939..3225508 100644 Binary files a/tests/e2e/test_diffusion_ref/expected_sdxl_dpo_lora.png and b/tests/e2e/test_diffusion_ref/expected_sdxl_dpo_lora.png differ diff --git a/tests/e2e/test_diffusion_ref/expected_sdxl_multi_loras.png b/tests/e2e/test_diffusion_ref/expected_sdxl_multi_loras.png index 035c81c..42d6361 100644 Binary files a/tests/e2e/test_diffusion_ref/expected_sdxl_multi_loras.png and b/tests/e2e/test_diffusion_ref/expected_sdxl_multi_loras.png differ