From 5fdc63b3c964e676ea459efa4854252fa0157e41 Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Thu, 3 Aug 2023 13:57:18 -0400 Subject: [PATCH] Pin model version used in unit tests. --- .../lora/injection/test_stable_diffusion_v1.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/tests/invoke_training/lora/injection/test_stable_diffusion_v1.py b/tests/invoke_training/lora/injection/test_stable_diffusion_v1.py index d6b6fd86..1e815ff1 100644 --- a/tests/invoke_training/lora/injection/test_stable_diffusion_v1.py +++ b/tests/invoke_training/lora/injection/test_stable_diffusion_v1.py @@ -12,7 +12,10 @@ def test_inject_lora_into_unet_sd1_smoke(): """Smoke test of inject_lora_into_unet_sd1(...) on full SD 1.5 model.""" unet = UNet2DConditionModel.from_pretrained( - "runwayml/stable-diffusion-v1-5", subfolder="unet", local_files_only=True + "runwayml/stable-diffusion-v1-5", + subfolder="unet", + local_files_only=True, + revision="c9ab35ff5f2c362e9e22fbafe278077e196057f0", ) lora_layers = inject_lora_into_unet_sd1(unet) @@ -29,7 +32,10 @@ def test_inject_lora_into_unet_sd1_smoke(): def test_convert_lora_state_dict_to_kohya_format_sd1_smoke(): """Smoke test of convert_lora_state_dict_to_kohya_format_sd1(...) with full SD 1.5 model.""" unet = UNet2DConditionModel.from_pretrained( - "runwayml/stable-diffusion-v1-5", subfolder="unet", local_files_only=True + "runwayml/stable-diffusion-v1-5", + subfolder="unet", + local_files_only=True, + revision="c9ab35ff5f2c362e9e22fbafe278077e196057f0", ) lora_layers = inject_lora_into_unet_sd1(unet)