Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
diffusers
Commits
f05d75c0
Unverified
Commit
f05d75c0
authored
Nov 06, 2023
by
Patrick von Platen
Committed by
GitHub
Nov 06, 2023
Browse files
[Custom Pipelines] Make sure that community pipelines can use repo revision (#5659)
fix custom pipelines
parent
aec3de8b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
1 deletion
+6
-1
src/diffusers/pipelines/pipeline_utils.py
src/diffusers/pipelines/pipeline_utils.py
+6
-1
No files found.
src/diffusers/pipelines/pipeline_utils.py
View file @
f05d75c0
...
...
@@ -353,13 +353,18 @@ def _get_pipeline_class(
else
:
file_name
=
CUSTOM_PIPELINE_FILE_NAME
if
repo_id
is
not
None
and
hub_revision
is
not
None
:
# if we load the pipeline code from the Hub
# make sure to overwrite the `revison`
revision
=
hub_revision
return
get_class_from_dynamic_module
(
custom_pipeline
,
module_file
=
file_name
,
class_name
=
class_name
,
repo_id
=
repo_id
,
cache_dir
=
cache_dir
,
revision
=
revision
if
hub_revision
is
None
else
hub_revision
,
revision
=
revision
,
)
if
class_obj
!=
DiffusionPipeline
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment