We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent df60b35 commit 1a4db89Copy full SHA for 1a4db89
src/diffusers/pipelines/pipeline_utils.py
@@ -353,13 +353,18 @@ def _get_pipeline_class(
353
else:
354
file_name = CUSTOM_PIPELINE_FILE_NAME
355
356
+ if repo_id is not None and hub_revision is not None:
357
+ # if we load the pipeline code from the Hub
358
+ # make sure to overwrite the `revison`
359
+ revision = hub_revision
360
+
361
return get_class_from_dynamic_module(
362
custom_pipeline,
363
module_file=file_name,
364
class_name=class_name,
365
repo_id=repo_id,
366
cache_dir=cache_dir,
- revision=revision if hub_revision is None else hub_revision,
367
+ revision=revision,
368
)
369
370
if class_obj != DiffusionPipeline:
0 commit comments