Skip to content

Commit 9ad1470

Browse files
committed
up
1 parent bf99ab2 commit 9ad1470

File tree

2 files changed

+1121
-975
lines changed

2 files changed

+1121
-975
lines changed

src/diffusers/pipelines/modular_pipeline.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ class ComponentSpec:
144144
name: str
145145
type_hint: Type
146146
description: Optional[str] = None
147-
default: Any = None # you can create a default component if it is a stateless class like scheduler, guider or image processor
147+
obj: Any = None # you can create a default component if it is a stateless class like scheduler, guider or image processor
148148
default_class_name: Union[str, List[str], Tuple[str, str]] = None # Either "class_name" or ["module", "class_name"]
149149
default_repo: Optional[Union[str, List[str]]] = None # either "repo" or ["repo", "subfolder"]
150150

@@ -185,6 +185,16 @@ def format_inputs_short(inputs):
185185
186186
Returns:
187187
str: Formatted string of input parameters
188+
189+
Example:
190+
>>> inputs = [
191+
... InputParam(name="prompt", required=True),
192+
... InputParam(name="image", required=True),
193+
... InputParam(name="guidance_scale", required=False, default=7.5),
194+
... InputParam(name="num_inference_steps", required=False, default=50)
195+
... ]
196+
>>> format_inputs_short(inputs)
197+
'prompt, image, guidance_scale=7.5, num_inference_steps=50'
188198
"""
189199
required_inputs = [param for param in inputs if param.required]
190200
optional_inputs = [param for param in inputs if not param.required]
@@ -367,13 +377,13 @@ def description(self) -> str:
367377
raise NotImplementedError("description method must be implemented in subclasses")
368378

369379
@property
370-
def components(self) -> List[ComponentSpec]:
380+
def expected_components(self) -> List[ComponentSpec]:
371381
return []
372382

373383
@property
374-
def configs(self) -> List[ConfigSpec]:
384+
def expected_configs(self) -> List[ConfigSpec]:
375385
return []
376-
386+
377387

378388
# YiYi TODO: can we combine inputs and intermediates_inputs? the difference is inputs are immutable
379389
@property

0 commit comments

Comments
 (0)