"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "03e51269782bd79e453232d1d3417c77e5412819"
Unverified Commit f05d75c0 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[Custom Pipelines] Make sure that community pipelines can use repo revision (#5659)

fix custom pipelines
parent aec3de8b
...@@ -353,13 +353,18 @@ def _get_pipeline_class( ...@@ -353,13 +353,18 @@ def _get_pipeline_class(
else: else:
file_name = CUSTOM_PIPELINE_FILE_NAME 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( return get_class_from_dynamic_module(
custom_pipeline, custom_pipeline,
module_file=file_name, module_file=file_name,
class_name=class_name, class_name=class_name,
repo_id=repo_id, repo_id=repo_id,
cache_dir=cache_dir, cache_dir=cache_dir,
revision=revision if hub_revision is None else hub_revision, revision=revision,
) )
if class_obj != DiffusionPipeline: if class_obj != DiffusionPipeline:
......
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