update model type

Signed-off-by: binliu <binliu@nvidia.com>

vista3d_model.py

@@ -13,10 +13,11 @@ class VISTA3DModel(PreTrainedModel):
 
     def __init__(self, config):
         super().__init__(config)
-        self.network = monai.networks.nets.vista3d132(
-            encoder_embed_dim=config.encoder_embed_dim,
-            in_channels=config.input_channels,
-        )
+        if config.model_type == "VISTA3D":
+            self.network = monai.networks.nets.vista3d132(
+                encoder_embed_dim=config.encoder_embed_dim,
+                in_channels=config.input_channels,
+            )
 
     def forward(self, input):
         return self.network(input)