diff --git a/.github/workflows/blossom-ci.yml b/.github/workflows/blossom-ci.yml index 4871ac9d..e2cc5fbb 100644 --- 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 index 50261b2c..55ab14b5 100644 --- 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 index 6b597a6c..1f87e8ee 100644 --- 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 index 923cc380..80e974b9 100644 --- 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",