Skip to content

Conversation

@RUFFY-369
Copy link

@RUFFY-369 RUFFY-369 commented Dec 26, 2025

Changes made on top of PR #488

@RUFFY-369 RUFFY-369 requested a review from eliteprox December 28, 2025 20:34
Copy link
Collaborator

@eliteprox eliteprox left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm!

eliteprox

This comment was marked as duplicate.

# Add new nodes as needed.

streamdiffusion:
script: /workspace/ComfyUI/custom_nodes/ComfyUI-StreamDiffusion/scripts/build_tensorrt_engines.py
Copy link
Collaborator

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.

Copy link
Collaborator

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

Copy link
Collaborator

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

979fcef

Copy link
Author

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.

@eliteprox eliteprox merged commit b0c4727 into feat/custom-nodes-config Dec 28, 2025
@eliteprox eliteprox deleted the prakarsh1.1/feat/custom-nodes-config branch December 28, 2025 21:29
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.

3 participants