diff --git a/src/diffusers/modular_pipelines/modular_pipeline_utils.py b/src/diffusers/modular_pipelines/modular_pipeline_utils.py index ced059551f..868c091060 100644 --- a/src/diffusers/modular_pipelines/modular_pipeline_utils.py +++ b/src/diffusers/modular_pipelines/modular_pipeline_utils.py @@ -41,6 +41,16 @@ class InsertableOrderedDict(OrderedDict): # Return self for method chaining return self + + def __repr__(self): + if not self: + return "InsertableOrderedDict()" + + items = [] + for i, (key, value) in enumerate(self.items()): + items.append(f"{i}: ({repr(key)}, {repr(value)})") + + return "InsertableOrderedDict([\n " + ",\n ".join(items) + "\n])" # YiYi TODO: