Skip to content

Commit 29de29f

Browse files
committed
add node_utils
1 parent 72e1b74 commit 29de29f

File tree

2 files changed

+355
-1
lines changed

2 files changed

+355
-1
lines changed

src/diffusers/modular_pipelines/modular_pipeline.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,6 @@ def from_pretrained(
303303
block_kwargs = {
304304
name: kwargs.pop(name) for name in kwargs if name in expected_kwargs or name in optional_kwargs
305305
}
306-
print(f"block_kwargs: {block_kwargs}")
307306

308307
return block_cls(**block_kwargs)
309308

@@ -749,6 +748,8 @@ def expected_configs(self):
749748

750749
@property
751750
def required_inputs(self) -> List[str]:
751+
if None not in self.block_trigger_inputs:
752+
return []
752753
first_block = next(iter(self.blocks.values()))
753754
required_by_all = set(getattr(first_block, "required_inputs", set()))
754755

@@ -763,6 +764,8 @@ def required_inputs(self) -> List[str]:
763764
# intermediate_inputs is by default required, unless you manually handle it inside the block
764765
@property
765766
def required_intermediates_inputs(self) -> List[str]:
767+
if None not in self.block_trigger_inputs:
768+
return []
766769
first_block = next(iter(self.blocks.values()))
767770
required_by_all = set(getattr(first_block, "required_intermediates_inputs", set()))
768771

0 commit comments

Comments
 (0)