fix(plugin): default_value to be optional in streaming image input

This commit is contained in:
bennykok 2024-03-02 12:22:17 -08:00
parent 3df549c25c
commit 011d36edce

View File

@ -44,7 +44,7 @@ class ComfyDeployWebscoketImageInput:
return True
def run(self, input_id, default_value, seed, client_id):
def run(self, input_id, seed, default_value=None ,client_id=None):
# print(streaming_prompt_metadata[client_id].inputs)
if client_id in streaming_prompt_metadata and input_id in streaming_prompt_metadata[client_id].inputs:
if isinstance(streaming_prompt_metadata[client_id].inputs[input_id], Image.Image):