python vision_export_onnx.py转化异常，出现TypeError: _adjust_pos_cache() missing 1 required positional argument: 'device'
python vision_export_onnx.py
/home/test/miniforge3/envs/RKLLM-Toolkit/lib/python3.8/site-packages/torch/cuda/init.py:611: UserWarning: Can't initialize NVML
warnings.warn("Can't initialize NVML")
Traceback (most recent call last):
File "vision_export_onnx.py", line 8, in
origin_model = AutoModelForCausalLM.from_pretrained(
File "/home/test/miniforge3/envs/RKLLM-Toolkit/lib/python3.8/site-packages/transformers/models/auto/auto_factory.py", line 559, in from_pretrained
return model_class.from_pretrained(
File "/home/test/miniforge3/envs/RKLLM-Toolkit/lib/python3.8/site-packages/transformers/modeling_utils.py", line 3880, in from_pretrained
model = cls(config, *model_args, **model_kwargs)
File "/home/test/.cache/huggingface/modules/transformers_modules/MiniCPM-V-2_6/modeling_minicpmv.py", line 29, in init
self.resampler = self.init_resampler(self.embed_dim, self.vision_dim)
File "/home/test/.cache/huggingface/modules/transformers_modules/MiniCPM-V-2_6/modeling_minicpmv.py", line 52, in init_resampler
return Resampler(
File "/home/test/.cache/huggingface/modules/transformers_modules/MiniCPM-V-2_6/resampler.py", line 107, in init
self._adjust_pos_cache([32,32])
TypeError: _adjust_pos_cache() missing 1 required positional argument: 'device'