Merge remote-tracking branch 'origin/dev' into maisi-trt

Project-MONAI · Oct 30, 2024 · 76f8fb6 · 76f8fb6
2 parents e827f5c + 7a5e69c
commit 76f8fb6
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 5 deletions.
diff --git a/.github/workflows/blossom-ci.yml b/.github/workflows/blossom-ci.yml
@@ -45,8 +45,7 @@ jobs:
     if: |
       github.event.comment.body == '/build' &&
       (
-        github.actor == 'pxLi' ||
-        github.actor == 'YanxuanLiu' ||
+        github.actor == 'wendell-hom' ||
         github.actor == 'wyli' ||
         github.actor == 'Nic-Ma' ||
         github.actor == 'yiheng-wang-nv' ||

diff --git a/models/model_info.json b/models/model_info.json
@@ -1782,5 +1782,9 @@
     "vista3d_v0.5.4": {
         "checksum": "05c03492449cad23267f1b97fd39c4153edd58b6",
         "source": "https://api.ngc.nvidia.com/v2/models/nvidia/monaihosting/vista3d/versions/0.5.4/files/vista3d_v0.5.4.zip"
+    },
+    "vista3d_v0.5.5": {
+        "checksum": "a4509e12c97042a86173c6bbd5fcfc0d2620c0ea",
+        "source": "https://api.ngc.nvidia.com/v2/models/nvidia/monaihosting/vista3d/versions/0.5.5/files/vista3d_v0.5.5.zip"
     }
 }
diff --git a/models/vista3d/configs/inference_trt.json b/models/vista3d/configs/inference_trt.json
@@ -1,4 +1,5 @@
 {
+    "base_path": null,
     "+imports": [
         "$from monai.networks import trt_compile"
     ],
@@ -8,11 +9,11 @@
         "dynamic_batchsize": "$[1, @inferer#sw_batch_size, @inferer#sw_batch_size]"
     },
     "network_dev": "$@network_def.to(@device)",
-    "encoder": "$trt_compile(@network_dev, @bundle_root + '/models/model.pt', args=@network_trt_args, submodule=['image_encoder.encoder'])",
+    "encoder": "$trt_compile(@network_dev, @bundle_root + '/models/model.pt' if not @base_path else @base_path, args=@network_trt_args, submodule=['image_encoder.encoder'])",
     "head_trt_args": {
         "dynamic_batchsize": "$[1, 1, @max_prompt_size]",
         "fallback": "$True"
     },
-    "head": "$trt_compile(@network_dev, @bundle_root + '/models/model.pt', args=@head_trt_args, submodule=['class_head']) if @head_trt_enabled else @network_dev",
+    "head": "$trt_compile(@network_dev, @bundle_root + '/models/model.pt' if not @base_path else @base_path, args=@head_trt_args, submodule=['class_head']) if @head_trt_enabled else @network_dev",
     "network": "$None if @encoder is None else @head"
 }
diff --git a/models/vista3d/configs/metadata.json b/models/vista3d/configs/metadata.json
@@ -1,7 +1,8 @@
 {
     "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20240725.json",
-    "version": "0.5.4",
+    "version": "0.5.5",
     "changelog": {
+        "0.5.5": "add arg for trt compiler base path",
         "0.5.4": "add undefined label prompt check",
         "0.5.3": "update readme",
         "0.5.2": "fix eval issue",