Skip to content

feat: make TensorRT export script configurable #14

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

Merged
merged 1 commit into from
May 20, 2025

Conversation

rickstaa
Copy link
Contributor

@rickstaa rickstaa commented Apr 3, 2025

This pull request makes the TensorRT engine export script configurable by adding CLI arguments for the output path (--trt-path), ONNX model path (--onnx-path), and precision (--use-fp16). These changes improve usability for DepthAnything
engines.

This commit makes the TensorRT engine export script configurable by adding
CLI arguments for the output path (--trt-path), ONNX model path (--onnx-path),
and precision (--use-fp16). These changes improve usability for DepthAnything
engines.
@rickstaa rickstaa force-pushed the feature/add-export-trt-args branch from ee7f75d to 1eb458d Compare April 3, 2025 23:17
@rickstaa rickstaa marked this pull request as ready for review April 3, 2025 23:21
@yuvraj108c yuvraj108c merged commit 117922a into yuvraj108c:master May 20, 2025
1 check passed
@yuvraj108c
Copy link
Owner

Thanks!

@rickstaa rickstaa deleted the feature/add-export-trt-args branch May 20, 2025 08:19
rickstaa added a commit to yondonfu/comfystream that referenced this pull request May 20, 2025
We previously used a fork of
https://github.com/yuvraj108c/ComfyUI-Depth-Anything-Tensorrt because the original
script lacked support for parameters needed to create multiple TensorRT engines.
Now that yuvraj108c/ComfyUI-Depth-Anything-Tensorrt#14
has been merged upstream, this workaround is no longer necessary. We can safely
return to the main repository.
@rickstaa rickstaa restored the feature/add-export-trt-args branch May 20, 2025 18:04
eliteprox pushed a commit to livepeer/comfystream that referenced this pull request May 20, 2025
Now that yuvraj108c/ComfyUI-Depth-Anything-Tensorrt#14
has been merged upstream, this workaround is no longer necessary. We can safely
return to the main repository.
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