1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

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

fix custom pipelines
This commit is contained in:
Patrick von Platen
2023-11-06 15:11:48 +01:00
committed by GitHub
parent aec3de8bdb
commit f05d75c076

View File

@@ -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: