Add TE import guard in qwen2vl vision module

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

Author: chtruong814

nemo/collections/vlm/qwen2vl/model/vision.py

@@ -32,6 +32,12 @@
 from megatron.core.utils import WrappedTensor, deprecate_inference_params, make_viewless_tensor
 from torch import Tensor
 
+try:
+    from megatron.core.extensions.transformer_engine import te_checkpoint
+    HAVE_TE = True
+except ImportError:
+    HAVE_TE = False
+
 
 class Qwen25VLVisionTransformerBlock(TransformerBlock):
     """