Pin model version used in unit tests.

invoke-ai · Aug 3, 2023 · 5fdc63b · 5fdc63b
1 parent 1cee5a7
commit 5fdc63b
Showing 1 changed file with 8 additions and 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
@@ -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)