Skip to content

feat: robustly handle engine version mismatch with automatic rebuilding - #27

Open
reaperhammer wants to merge 1 commit into
yuvraj108c:masterfrom
reaperhammer:fix-version-mismatch-loading
Open

feat: robustly handle engine version mismatch with automatic rebuilding#27
reaperhammer wants to merge 1 commit into
yuvraj108c:masterfrom
reaperhammer:fix-version-mismatch-loading

Conversation

@reaperhammer

Copy link
Copy Markdown
Contributor

Make the engine loading process self-healing. If TensorRT version/compilation settings in the environment change, loading the serialized engine (.trt) will fail with a PolygraphyException. Instead of crashing ComfyUI execution, this change catches the exception, deletes the outdated/incompatible engine file, and automatically compiles a compatible engine from the ONNX source.

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.

1 participant