TypeError

[allthemight]

python3 -u main.py --scenario Offline --model-path ${CHECKPOINT_PATH} --mlperf-conf mlperf.conf --user-conf user.conf --total-sample-count 24576 --dataset-path ${DATASET_PATH} --output-log-dir offline-logs --dtype float32 --device cuda:0 2>&1 | tee offline_performance_log.log
Traceback (most recent call last):
File "/home/fangtao/inference/language/llama2-70b/main.py", line 96, in
main()
File "/home/fangtao/inference/language/llama2-70b/main.py", line 69, in main
sut = sut_cls(
File "/home/fangtao/inference/language/llama2-70b/SUT.py", line 97, in init
self.data_object = Dataset(self.model_path, dataset_path=self.dataset_path, total_sample_count=total_sample_count)
File "/home/fangtao/inference/language/llama2-70b/dataset.py", line 30, in init
self.load_processed_dataset()
File "/home/fangtao/inference/language/llama2-70b/dataset.py", line 52, in load_processed_dataset
input_tokens = processed_data['tok_input']
File "/root/miniconda3/envs/llama2-70b/lib/python3.9/site-packages/pandas/core/frame.py", line 4078, in getitem
return self._get_item_cache(key)
File "/root/miniconda3/envs/llama2-70b/lib/python3.9/site-packages/pandas/core/frame.py", line 4639, in _get_item_cache
res = self._ixs(loc, axis=1)
File "/root/miniconda3/envs/llama2-70b/lib/python3.9/site-packages/pandas/core/frame.py", line 4010, in _ixs
col_mgr = self._mgr.iget(i)
File "/root/miniconda3/envs/llama2-70b/lib/python3.9/site-packages/pandas/core/internals/managers.py", line 1016, in iget
block = self.blocks[self.blknos[i]]
File "/root/miniconda3/envs/llama2-70b/lib/python3.9/site-packages/pandas/core/internals/managers.py", line 192, in blknos
self._rebuild_blknos_and_blklocs()
File "internals.pyx", line 761, in pandas._libs.internals.BlockManager._rebuild_blknos_and_blklocs
TypeError: Cannot convert numpy.ndarray to numpy.ndarray