corrects single file path validation logic

Author: andrew-w-ross

src/diffusers/loaders/single_file_utils.py

@@ -409,7 +409,10 @@ def is_valid_url(url):
 
 
 def _is_single_file_path_or_url(pretrained_model_name_or_path):
-    if not os.path.isfile(pretrained_model_name_or_path) or not is_valid_url(pretrained_model_name_or_path):
+    if os.path.isfile(pretrained_model_name_or_path):
+        return True
+
+    if not is_valid_url(pretrained_model_name_or_path):
         return False
 
     repo_id, weight_name = _extract_repo_id_and_weights_name(pretrained_model_name_or_path)

tests/modular_pipelines/test_modular_pipelines_common.py

@@ -8,7 +8,7 @@
 from huggingface_hub import hf_hub_download
 
 import diffusers
-from diffusers import AutoModel, ComponentsManager, ModularPipeline, ModularPipelineBlocks
+from diffusers import AutoModel, ComponentsManager, ControlNetModel, ModularPipeline, ModularPipelineBlocks
 from diffusers.guiders import ClassifierFreeGuidance
 from diffusers.modular_pipelines.modular_pipeline_utils import (
     ComponentSpec,
@@ -727,6 +727,28 @@ def test_automodel_update_components(self):
         assert spec.pretrained_model_name_or_path == "hf-internal-testing/tiny-stable-diffusion-xl-pipe"
         assert spec.subfolder == "unet"
 
+    def test_load_components_loads_local_single_file_path(self, tmp_path):
+        pipe = ModularPipeline.from_pretrained("hf-internal-testing/tiny-stable-diffusion-xl-pipe")
+
+        local_ckpt_path = hf_hub_download(
+            repo_id="lllyasviel/ControlNet-v1-1",
+            filename="control_v11p_sd15_canny.pth",
+            local_dir=str(tmp_path),
+        )
+
+        pipe._component_specs["controlnet"] = ComponentSpec(
+            name="controlnet",
+            type_hint=ControlNetModel,
+            pretrained_model_name_or_path=local_ckpt_path,
+        )
+
+        pipe.load_components(names="controlnet")
+
+        assert pipe.controlnet is not None
+        assert isinstance(pipe.controlnet, ControlNetModel)
+        assert pipe._component_specs["controlnet"].pretrained_model_name_or_path == local_ckpt_path
+        assert getattr(pipe.controlnet, "_diffusers_load_id", None) not in (None, "null")
+
 
 class TestLoadComponentsSkipBehavior:
     def test_load_components_skips_already_loaded(self):