@@ -1257,8 +1257,8 @@ def tearDown(self):
12571257
12581258 def test_download_from_hub (self ):
12591259 ckpt_paths = [
1260- "https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.ckpt " ,
1261- "https://huggingface.co/WarriorMama777/OrangeMixs/blob/main/Models/AbyssOrangeMix/AbyssOrangeMix_base.ckpt " ,
1260+ "https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors " ,
1261+ "https://huggingface.co/WarriorMama777/OrangeMixs/blob/main/Models/AbyssOrangeMix/AbyssOrangeMix.safetensors " ,
12621262 ]
12631263
12641264 for ckpt_path in ckpt_paths :
@@ -1271,7 +1271,7 @@ def test_download_from_hub(self):
12711271 assert image_out .shape == (512 , 512 , 3 )
12721272
12731273 def test_download_local (self ):
1274- ckpt_filename = hf_hub_download ("runwayml/stable-diffusion-v1-5" , filename = "v1-5-pruned-emaonly.ckpt " )
1274+ ckpt_filename = hf_hub_download ("runwayml/stable-diffusion-v1-5" , filename = "v1-5-pruned-emaonly.safetensors " )
12751275 config_filename = hf_hub_download ("runwayml/stable-diffusion-v1-5" , filename = "v1-inference.yaml" )
12761276
12771277 pipe = StableDiffusionPipeline .from_single_file (
@@ -1285,7 +1285,7 @@ def test_download_local(self):
12851285 assert image_out .shape == (512 , 512 , 3 )
12861286
12871287 def test_download_ckpt_diff_format_is_same (self ):
1288- ckpt_path = "https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.ckpt "
1288+ ckpt_path = "https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors "
12891289
12901290 sf_pipe = StableDiffusionPipeline .from_single_file (ckpt_path )
12911291 sf_pipe .scheduler = DDIMScheduler .from_config (sf_pipe .scheduler .config )
@@ -1310,7 +1310,7 @@ def test_download_ckpt_diff_format_is_same(self):
13101310 def test_single_file_component_configs (self ):
13111311 pipe = StableDiffusionPipeline .from_pretrained ("runwayml/stable-diffusion-v1-5" )
13121312
1313- ckpt_path = "https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.ckpt "
1313+ ckpt_path = "https://huggingface.co/runwayml/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors "
13141314 single_file_pipe = StableDiffusionPipeline .from_single_file (ckpt_path , load_safety_checker = True )
13151315
13161316 for param_name , param_value in single_file_pipe .text_encoder .config .to_dict ().items ():
0 commit comments