Skip to content

Commit 9f9db3b

Browse files
committed
update
1 parent 974f67e commit 9f9db3b

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/pipelines/test_pipeline_utils.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -403,6 +403,12 @@ def test_sharded_mixed_variants_downloaded(self):
403403
model_filenames, variant_filenames = variant_compatible_siblings(filenames, variant=variant)
404404
assert all(variant in f if allowed_non_variant not in f else variant not in f for f in model_filenames)
405405

406+
def test_downloading_when_no_variant_exists(self):
407+
variant = "fp16"
408+
filenames = ["model.safetensors", "diffusion_pytorch_model.safetensors"]
409+
model_filenames, variant_filenames = variant_compatible_siblings(filenames, variant=variant)
410+
assert len(model_filenames) != 0
411+
406412

407413
class ProgressBarTests(unittest.TestCase):
408414
def get_dummy_components_image_generation(self):

0 commit comments

Comments
 (0)