Skip to content

added compatibility with several comfyUI workflows#772

Merged
zanllp merged 4 commits into
zanllp:mainfrom
ermanitu:comfyUI_improvements
Jan 14, 2025
Merged

added compatibility with several comfyUI workflows#772
zanllp merged 4 commits into
zanllp:mainfrom
ermanitu:comfyUI_improvements

Conversation

@ermanitu

Copy link
Copy Markdown
Contributor

First improvement:

The input to a KSampler model is not always from a LoadCheckPoint node, as it can be mediated by a LoadLora or similar. In these cases, IIB was failing and not continuing the execution of the function.

Note that in python "get" doesn't cause error if the key doesn't exists. Instead of this, it returns None. This not get properly the model used, but at least get the rest of the params.

Second improvement:

The code assumes the "positive" input of KSampler is the positive prompt, but may not be the case. Some flux.1 workflows have a node "Guidance" between the positive prompt and the KSampler.

If there is a FluxGuidance node then give an step back to get the prompt.

Images to test:

383215349-3f0c0128-e8ab-4c55-93f6-fb51c6a9eae3

ComfyUI_temp_sunbk_00002_

@ermanitu

Copy link
Copy Markdown
Contributor Author

New improvement in the loop for searching for KSampler node, because some of my sources has a key over 32. I thing is better to do the for loop over the data.keys(), and it's more efficient because only test existing nodes.

@zanllp zanllp merged commit a6925ab into zanllp:main Jan 14, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants