Closed yun189 closed 1 day ago
I have never encountered this error before. Maybe it's due to python 3.10...
请问您使用python3.10吗?还是其他版本?
我用3.9
我本地debug的时候3.8也用过 你直接改成3.8吧
好的,感谢您的回复。
非常感谢您的快速解答!用python3.8装了一遍您刚才提供的sh,还是报错:
File "inference_demo.py", line 2, in
对 这都是一些版本问题 cuda+torch+cudnn+python之间的版本兼容问题 还有一些还会涉及到你的nvidia驱动版本 比较常见
python inference_demo.py, error happened: Traceback (most recent call last): File "/mnt/sdb/MiCo/inference_demo.py", line 171, in
outputs = model.multimodal_encoder.generate(input_ids=init_input_ids,
File "/home/liran/miniforge3/envs/MiCo/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/home/liran/miniforge3/envs/MiCo/lib/python3.10/site-packages/transformers/generation/utils.py", line 1953, in generate
result = self._beam_search(
File "/home/liran/miniforge3/envs/MiCo/lib/python3.10/site-packages/transformers/generation/utils.py", line 3011, in _beam_search
model_kwargs["past_key_values"] = self._temporary_reorder_cache(
File "/home/liran/miniforge3/envs/MiCo/lib/python3.10/site-packages/transformers/generation/utils.py", line 2756, in _temporary_reorder_cache
past_key_values = self._reorder_cache(past_key_values, beam_idx)
File "/home/liran/miniforge3/envs/MiCo/lib/python3.10/site-packages/transformers/generation/utils.py", line 699, in _reorder_cache
raise NotImplementedError(
NotImplementedError: Make sure that a
_reorder_cache
function is correctly implemented in model.bert to enable beam search for <class 'model.bert.BertForMaskedLM'> what can I do for it?