-
Notifications
You must be signed in to change notification settings - Fork 6
Fix:streamdiffusion custom node build, dependencies conflict issue #579
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fix:streamdiffusion custom node build, dependencies conflict issue #579
Conversation
eliteprox
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
lgtm!
| # Add new nodes as needed. | ||
|
|
||
| streamdiffusion: | ||
| script: /workspace/ComfyUI/custom_nodes/ComfyUI-StreamDiffusion/scripts/build_tensorrt_engines.py |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This is neat, does it support arguments for other scripts like build_trt.py?
We will need to cover the case when an image is built using nodes.yaml, and entrypoint.sh is ran with --build-engines, Just need a basic check to not attempt build if the custom nodes folder or script doesn't exist.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We can finalize that in #488 though, will go ahead and merge this one
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can cherry-pick this one to guard the other engine builds or refactor to your script
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This is neat, does it support arguments for other scripts like build_trt.py?
We will need to cover the case when an image is built using nodes.yaml, and entrypoint.sh is ran with
--build-engines, Just need a basic check to not attempt build if the custom nodes folder or script doesn't exist.
The [entrypoint.sh]already checks for the existence of build_targets.yaml, the required folder, and the script before attempting StreamDiffusion engine builds (see lines 142–144 and the Python block around line 150+). It skips the build if any are missing, just like the new logic for other nodes which you referred for cherry picking.
Changes made on top of PR #488