fix

Author: yiyixuxu

src/diffusers/modular_pipelines/modular_pipeline.py

@@ -1351,8 +1351,8 @@ def get_inputs(self):
                 input_param.required = False
         return combined_inputs
 
-    # Copied from SequentialPipelineBlocks
     @property
+    # Copied from diffusers.modular_pipelines.modular_pipeline.SequentialPipelineBlocks.inputs
     def inputs(self):
         return self.get_inputs()
 
@@ -1366,7 +1366,7 @@ def intermediates_inputs(self):
                 intermediates.append(loop_intermediate_input)
         return intermediates
 
-    # Copied from SequentialPipelineBlocks
+    # modified from SequentialPipelineBlocks
     def get_intermediates_inputs(self):
         inputs = []
         outputs = set()
@@ -1429,7 +1429,6 @@ def intermediates_outputs(self) -> List[str]:
         return combined_outputs
 
     # YiYi TODO: this need to be thought about more
-    # copied from SequentialPipelineBlocks
     @property
     def outputs(self) -> List[str]:
         return next(reversed(self.blocks.values())).intermediates_outputs

src/diffusers/modular_pipelines/modular_pipeline_utils.py

@@ -80,7 +80,7 @@ class ComponentSpec:
     name: Optional[str] = None
     type_hint: Optional[Type] = None
     description: Optional[str] = None
-    config: Optional[FrozenDict[str, Any]] = None
+    config: Optional[FrozenDict] = None
     # YiYi Notes: should we change it to pretrained_model_name_or_path for consistency? a bit long for a field name
     repo: Optional[Union[str, List[str]]] = field(default=None, metadata={"loading": True})
     subfolder: Optional[str] = field(default=None, metadata={"loading": True})