Why am I getting this error?

[Shehjad-Ishan]

python3 tree_predict.py --prompt="[an owl [a wing, an eye]]" --threshold=0.15 --image_encoder_engine=../data/owl_image_encoder_patch32.engine
2024-11-19 16:58:45.021604: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:485] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2024-11-19 16:58:45.090744: E external/local_xla/xla/stream_executor/cuda/cuda_dnn.cc:8454] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
2024-11-19 16:58:45.109901: E external/local_xla/xla/stream_executor/cuda/cuda_blas.cc:1452] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
2024-11-19 16:58:45.287939: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
/home/sigmind/Downloads/[/mnt/DockerRuntime/miniconda]/lib/python3.10/site-packages/torch/functional.py:534: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3595.)
return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]
/media/sigmind/URSTP_HDD1416/nanoowl/nanoowl/image_preprocessor.py:71: UserWarning: The given NumPy array is not writable, and PyTorch does not support non-writable tensors. This means writing to this tensor will result in undefined behavior. You may want to copy the array to protect its data or make it writable before converting it to a tensor. This type of warning will be suppressed for the rest of this program. (Triggered internally at ../torch/csrc/utils/tensor_numpy.cpp:206.)
image = torch.from_numpy(np.asarray(image))
[11/19/2024-16:59:01] [TRT] [W] The enqueue() method has been deprecated when used with engines built from a network created with NetworkDefinitionCreationFlag::kEXPLICIT_BATCH flag. Please use enqueueV2() instead.
[11/19/2024-16:59:01] [TRT] [W] Also, the batchSize argument passed into this function has no effect on changing the input shapes. Please use setBindingDimensions() function to change input shapes instead.
Traceback (most recent call last):
File "/media/sigmind/URSTP_HDD1416/nanoowl/examples/tree_predict.py", line 51, in
output = predictor.predict(
File "/home/sigmind/Downloads/[/mnt/DockerRuntime/miniconda]/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 116, in decorate_context
return func(*args, **kwargs)
File "/media/sigmind/URSTP_HDD1416/nanoowl/nanoowl/tree_predictor.py", line 141, in predict
owl_node_output = self.owl_predictor.decode(
File "/media/sigmind/URSTP_HDD1416/nanoowl/nanoowl/owl_predictor.py", line 314, in decode
input_indices=input_indices[mask]
IndexError: too many indices for tensor of dimension 2