@@ -1257,8 +1257,8 @@ def tearDown(self):
1257
1257
1258
1258
def test_download_from_hub (self ):
1259
1259
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 " ,
1262
1262
]
1263
1263
1264
1264
for ckpt_path in ckpt_paths :
@@ -1271,7 +1271,7 @@ def test_download_from_hub(self):
1271
1271
assert image_out .shape == (512 , 512 , 3 )
1272
1272
1273
1273
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 " )
1275
1275
config_filename = hf_hub_download ("runwayml/stable-diffusion-v1-5" , filename = "v1-inference.yaml" )
1276
1276
1277
1277
pipe = StableDiffusionPipeline .from_single_file (
@@ -1285,7 +1285,7 @@ def test_download_local(self):
1285
1285
assert image_out .shape == (512 , 512 , 3 )
1286
1286
1287
1287
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 "
1289
1289
1290
1290
sf_pipe = StableDiffusionPipeline .from_single_file (ckpt_path )
1291
1291
sf_pipe .scheduler = DDIMScheduler .from_config (sf_pipe .scheduler .config )
@@ -1310,7 +1310,7 @@ def test_download_ckpt_diff_format_is_same(self):
1310
1310
def test_single_file_component_configs (self ):
1311
1311
pipe = StableDiffusionPipeline .from_pretrained ("runwayml/stable-diffusion-v1-5" )
1312
1312
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 "
1314
1314
single_file_pipe = StableDiffusionPipeline .from_single_file (ckpt_path , load_safety_checker = True )
1315
1315
1316
1316
for param_name , param_value in single_file_pipe .text_encoder .config .to_dict ().items ():
0 commit comments