KeyError: ('torch._utils', '_rebuild_parameter') when convert CodeShell-7B-Chat-int4

[workmanihcj]

when use convert.py convert CodeShell-7B-Chat-int4\pytorch_model.bin output below error, please help me find reason,thanks!

(base) D:\llama_tools>Python convert.py models/CodeShell-7B-Chat-int4/
Loading model file models\CodeShell-7B-Chat-int4\pytorch_model.bin
Traceback (most recent call last):
File "D:\llama_tools\convert.py", line 1229, in
main()
File "D:\llama_tools\convert.py", line 1162, in main
model_plus = load_some_model(args.model)
File "D:\llama_tools\convert.py", line 1077, in load_some_model
models_plus.append(lazy_load_file(path))
File "D:\llama_tools\convert.py", line 751, in lazy_load_file
return lazy_load_torch_file(fp, path)
File "D:\llama_tools\convert.py", line 707, in lazy_load_torch_file
model = unpickler.load()
File "D:\llama_tools\convert.py", line 696, in find_class
return self.CLASSES[(module, name)]
KeyError: ('torch._utils', '_rebuild_parameter')