Fix iMF/pMF inference by filtering unsupported pipeline kwargs

model_loader.py

@@ -334,6 +334,18 @@ def default_class_label_for_pipe(pipe: DiffusionPipeline, profile: ModelProfile)
     return synonyms[0] if synonyms else profile.default_class_label
 
 
+def _filter_call_kwargs(pipe: DiffusionPipeline, call_kwargs: dict[str, Any]) -> dict[str, Any]:
+    """Drop kwargs that the pipeline __call__ does not accept (e.g. height/width for iMF/pMF)."""
+    try:
+        params = inspect.signature(pipe.__call__).parameters
+    except (TypeError, ValueError):
+        return call_kwargs
+    if any(param.kind == inspect.Parameter.VAR_KEYWORD for param in params.values()):
+        return call_kwargs
+    accepted = set(params.keys())
+    return {key: value for key, value in call_kwargs.items() if key in accepted}
+
+
 def run_inference(
     profile: ModelProfile,
     pipe: DiffusionPipeline,
@@ -376,4 +388,4 @@ def run_inference(
             call_kwargs["height"] = height
             call_kwargs["width"] = width
 
-    return pipe(**call_kwargs).images[0]
+    return pipe(**_filter_call_kwargs(pipe, call_kwargs)).images[0]