Prebuilt 448x560 TensorRT engines: Jetson Orin (TRT 10.7) and RTX 50-series (TRT 10.16)

.gitattributes

@@ -34,3 +34,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 da2_metric_hypersim_vits_364x448_trt100700_sm87.engine filter=lfs diff=lfs merge=lfs -text
+da2_metric_hypersim_vits_448x560_trt100700_sm87.engine filter=lfs diff=lfs merge=lfs -text
+da2_metric_hypersim_vits_448x560_trt101601_sm120.engine filter=lfs diff=lfs merge=lfs -text

README.md

@@ -16,8 +16,9 @@ Depth Anything V2 Small, metric, Hypersim (indoor, max depth 20 m), converted fr
   (converted by `tools/convert_weights.py`; includes the unused ImageNet classifier head candle's dinov2 loads).
 - `da2_metric_hypersim_vits.onnx`: an ONNX export taking any input size that is a multiple of 14, for TensorRT
   (made by the crate's `tools/export_onnx.py`). `da2_metric_hypersim_vits_364x448.onnx` is the older fixed-size one.
-- `da2_metric_hypersim_vits_364x448_trt100700_sm87.engine`: that ONNX as an fp16 TensorRT 10.7 engine, built on a
-  Jetson AGX Orin (sm_87, JetPack 6 / L4T R36.3), so Orins skip the minutes-long engine build. Other TensorRT
-  versions or GPUs reject it and the crate builds its own from the ONNX.
+- Prebuilt fp16 TensorRT engines for a 448x560 input (dimos's default), so these GPUs skip the minutes-long engine build:
+  `..._448x560_trt100700_sm87.engine` (TensorRT 10.7, Jetson AGX Orin, JetPack 6 / L4T R36.3) and
+  `..._448x560_trt101601_sm120.engine` (TensorRT 10.16, RTX 50-series). Other TensorRT versions, GPUs or sizes reject
+  them and the crate builds its own from the ONNX. (`..._364x448_trt100700_sm87.engine` is an older fixed-size one.)
 
 The crate pins these files by sha256 in its `model.json`. Weights are Apache-2.0, as the upstream Small model.

da2_metric_hypersim_vits_448x560_trt100700_sm87.engine

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:be974849b8704bc795254e248610a0408371cbdafdab29e2411953b9b0e2b4f9
+size 55422620

da2_metric_hypersim_vits_448x560_trt101601_sm120.engine

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7b9defa44791e09faac986b8163b6fd7c822b78a6f02dc24a4d6c11d541e9147
+size 53640260