demo error

[7423cg]

(inpainting) cg@JJ:~/Desktop/AOT-GAN-for-Inpainting/src$ python demo.py --dir_image ./data/images --pre_train ./experiments/G0000000.pt --painter freeform
Traceback (most recent call last):
File "demo.py", line 110, in
demo(args)
File "demo.py", line 31, in demo
model.load_state_dict(torch.load(args.pre_train, map_location="gpu"))
File "/home/cg/anaconda3/envs/inpainting/lib/python3.8/site-packages/torch/serialization.py", line 815, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/home/cg/anaconda3/envs/inpainting/lib/python3.8/site-packages/torch/serialization.py", line 1018, in _legacy_load
return legacy_load(f)
File "/home/cg/anaconda3/envs/inpainting/lib/python3.8/site-packages/torch/serialization.py", line 904, in legacy_load
tar.extract('storages', path=tmpdir)
File "/home/cg/anaconda3/envs/inpainting/lib/python3.8/tarfile.py", line 2060, in extract
tarinfo = self.getmember(member)
File "/home/cg/anaconda3/envs/inpainting/lib/python3.8/tarfile.py", line 1782, in getmember
raise KeyError("filename %r not found" % name)
KeyError: "filename 'storages' not found"