Skip to content

Commit 03ca113

Browse files
DN6sayakpaul
andauthored
Update download diff format tests (#7831)
update Co-authored-by: Sayak Paul <[email protected]>
1 parent 3ffa7b4 commit 03ca113

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tests/pipelines/stable_diffusion/test_stable_diffusion.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)