[Modular] Fix for custom block kwargs (#12561)

DN6 · web-flow · commit df8dd778177c · 2025-10-31T00:14:24.000+05:30
update
diff --git a/src/diffusers/modular_pipelines/modular_pipeline.py b/src/diffusers/modular_pipelines/modular_pipeline.py
@@ -335,7 +335,7 @@ def from_pretrained(
         )
         expected_kwargs, optional_kwargs = block_cls._get_signature_keys(block_cls)
         block_kwargs = {
-            name: kwargs.pop(name) for name in kwargs if name in expected_kwargs or name in optional_kwargs
+            name: kwargs.get(name) for name in kwargs if name in expected_kwargs or name in optional_kwargs
         }
 
         return block_cls(**block_kwargs)

Original file line number	Diff line number	Diff line change
`@@ -335,7 +335,7 @@ def from_pretrained(`
`335`	`335`	`)`
`336`	`336`	`expected_kwargs, optional_kwargs = block_cls._get_signature_keys(block_cls)`
`337`	`337`	`block_kwargs = {`
`338`		`- name: kwargs.pop(name) for name in kwargs if name in expected_kwargs or name in optional_kwargs`
	`338`	`+ name: kwargs.get(name) for name in kwargs if name in expected_kwargs or name in optional_kwargs`
`339`	`339`	`}`
`340`	`340`
`341`	`341`	`return block_cls(**block_kwargs)`