up

Author: patrickvonplaten

tests/test_config.py

@@ -194,7 +194,7 @@ def test_load_ddim_from_pndm(self):
         logger = logging.get_logger("diffusers.configuration_utils")
 
         with CaptureLogger(logger) as cap_logger:
-            ddim = DDIMScheduler.from_config("runwayml/stable-diffusion-v1-5", subfolder="scheduler")
+            ddim = DDIMScheduler.from_config("CompVis/stable-diffusion-v1-4", subfolder="scheduler")
 
         assert ddim.__class__ == DDIMScheduler
         # no warning should be thrown
@@ -204,7 +204,7 @@ def test_load_pndm(self):
         logger = logging.get_logger("diffusers.configuration_utils")
 
         with CaptureLogger(logger) as cap_logger:
-            pndm = PNDMScheduler.from_config("runwayml/stable-diffusion-v1-5", subfolder="scheduler")
+            pndm = PNDMScheduler.from_config("CompVis/stable-diffusion-v1-4", subfolder="scheduler")
 
         assert pndm.__class__ == PNDMScheduler
         # no warning should be thrown