From d483042c093c27eb98f11c974906f84b508ba56c Mon Sep 17 00:00:00 2001 From: Sirshendu Ganguly Date: Mon, 13 Oct 2025 19:38:38 -0400 Subject: [PATCH] Add custom input structure support for sourceful models - Added check to detect sourceful models by AIR code prefix - When model starts with 'sourceful', use inputs.references structure - Maintains backward compatibility with referenceImages for other models - Supports new API format: inputs: {references: [image_urls]} --- modules/imageInference.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/modules/imageInference.py b/modules/imageInference.py index b07d2f5..3fef7e5 100644 --- a/modules/imageInference.py +++ b/modules/imageInference.py @@ -303,7 +303,11 @@ class txt2img: # Handle referenceImages if referenceImages is not None: - genConfig[0]["referenceImages"] = referenceImages + # Check if model is sourceful - use different structure + if runwareModel.startswith("sourceful"): + genConfig[0]["inputs"] = {"references": referenceImages} + else: + genConfig[0]["referenceImages"] = referenceImages # Handle providerSettings - extract provider name from model and merge with custom settings if providerSettings is not None: