Unverified Commit d1b2a1a9 authored by PENGUINLIONG's avatar PENGUINLIONG Committed by GitHub
Browse files

Fixed custom module importing on Windows (#5891)



* Fixed custom module importing on Windows

Windows use back slash and `os.path.join()` follows that convention.

* Apply suggestions from code review
Co-authored-by: default avatarLucain <lucainp@gmail.com>

* Update pipeline_utils.py

---------
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
Co-authored-by: default avatarLucain <lucainp@gmail.com>
parent 01782c22
...@@ -1688,7 +1688,8 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin): ...@@ -1688,7 +1688,8 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
if module_candidate is None or not isinstance(module_candidate, str): if module_candidate is None or not isinstance(module_candidate, str):
continue continue
candidate_file = os.path.join(component, module_candidate + ".py") # We compute candidate file path on the Hub. Do not use `os.path.join`.
candidate_file = f"{component}/{module_candidate}.py"
if candidate_file in filenames: if candidate_file in filenames:
custom_components[component] = module_candidate custom_components[component] = module_candidate
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment