update

Author: DN6

src/diffusers/pipelines/pipeline_loading_utils.py

@@ -154,7 +154,9 @@ def filter_model_files(filenames):
     if is_transformers_available():
         weight_names += [TRANSFORMERS_WEIGHTS_NAME, TRANSFORMERS_SAFE_WEIGHTS_NAME, TRANSFORMERS_FLAX_WEIGHTS_NAME]
 
-    return [f for f in filenames if any(f.endswith(wn) for wn in weight_names)]
+    allowed_extensions = [wn.split(".")[-1] for wn in weight_names]
+
+    return [f for f in filenames if any(f.endswith(extension) for extension in allowed_extensions)]
 
 
 def variant_compatible_siblings(filenames, variant=None, ignore_patterns=None) -> Union[List[os.PathLike], str]:

src/diffusers/pipelines/pipeline_utils.py

@@ -1432,6 +1432,7 @@ def download(cls, pretrained_model_name, **kwargs) -> Union[str, os.PathLike]:
             expected_components, _ = cls._get_signature_keys(pipeline_class)
             passed_components = [k for k in expected_components if k in kwargs]
 
+            # retrieve the names of the folders containing model weights
             model_folder_names = {
                 os.path.split(f)[0] for f in filter_model_files(filenames) if os.path.split(f)[0] in folder_names
             }