@@ -244,28 +244,30 @@ def test_load_ddim_from_pndm(self):
244244 logger = logging .get_logger ("diffusers.configuration_utils" )
245245
246246 with CaptureLogger (logger ) as cap_logger :
247- ddim = DDIMScheduler .from_config ("runwayml/ stable-diffusion-v1-5 " , subfolder = "scheduler" )
247+ ddim = DDIMScheduler .from_config ("hf-internal-testing/tiny- stable-diffusion-torch " , subfolder = "scheduler" )
248248
249249 assert ddim .__class__ == DDIMScheduler
250250 # no warning should be thrown
251251 assert cap_logger .out == ""
252252
253- def test_load_ddim_from_euler (self ):
253+ def test_load_euler_from_pndm (self ):
254254 logger = logging .get_logger ("diffusers.configuration_utils" )
255255
256256 with CaptureLogger (logger ) as cap_logger :
257- euler = EulerDiscreteScheduler .from_config ("runwayml/stable-diffusion-v1-5" , subfolder = "scheduler" )
257+ euler = EulerDiscreteScheduler .from_config (
258+ "hf-internal-testing/tiny-stable-diffusion-torch" , subfolder = "scheduler"
259+ )
258260
259261 assert euler .__class__ == EulerDiscreteScheduler
260262 # no warning should be thrown
261263 assert cap_logger .out == ""
262264
263- def test_load_ddim_from_euler_ancestral (self ):
265+ def test_load_euler_ancestral_from_pndm (self ):
264266 logger = logging .get_logger ("diffusers.configuration_utils" )
265267
266268 with CaptureLogger (logger ) as cap_logger :
267269 euler = EulerAncestralDiscreteScheduler .from_config (
268- "runwayml/ stable-diffusion-v1-5 " , subfolder = "scheduler"
270+ "hf-internal-testing/tiny- stable-diffusion-torch " , subfolder = "scheduler"
269271 )
270272
271273 assert euler .__class__ == EulerAncestralDiscreteScheduler
@@ -276,7 +278,7 @@ def test_load_pndm(self):
276278 logger = logging .get_logger ("diffusers.configuration_utils" )
277279
278280 with CaptureLogger (logger ) as cap_logger :
279- pndm = PNDMScheduler .from_config ("runwayml/ stable-diffusion-v1-5 " , subfolder = "scheduler" )
281+ pndm = PNDMScheduler .from_config ("hf-internal-testing/tiny- stable-diffusion-torch " , subfolder = "scheduler" )
280282
281283 assert pndm .__class__ == PNDMScheduler
282284 # no warning should be thrown
0 commit comments