We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 974f67e commit 9f9db3bCopy full SHA for 9f9db3b
tests/pipelines/test_pipeline_utils.py
@@ -403,6 +403,12 @@ def test_sharded_mixed_variants_downloaded(self):
403
model_filenames, variant_filenames = variant_compatible_siblings(filenames, variant=variant)
404
assert all(variant in f if allowed_non_variant not in f else variant not in f for f in model_filenames)
405
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
+
412
413
class ProgressBarTests(unittest.TestCase):
414
def get_dummy_components_image_generation(self):
0 commit comments