OOTDiffusion icon indicating copy to clipboard operation
OOTDiffusion copied to clipboard

pickle_module.load(f, **pickle_load_args)

Open xiyuxiaoxiao opened this issue 1 year ago • 3 comments

python run_ootd.py --model_path /root/OOTDiffusion/run/examples/model/01008_00.jpg --cloth_path /root/OOTDiffusion/run/examples/garment/00055_00.jpg --scale 2.0 --sample 4

-: Traceback (most recent call last): File "/root/OOTDiffusion/run/run_ootd.py", line 29, in openpose_model = OpenPose(args.gpu_id) File "/root/OOTDiffusion/preprocess/openpose/run_openpose.py", line 33, in init self.preprocessor = OpenposeDetector() File "/root/OOTDiffusion/preprocess/openpose/annotator/openpose/init.py", line 63, in init self.body_estimation = Body(body_modelpath) File "/root/OOTDiffusion/preprocess/openpose/annotator/openpose/body.py", line 26, in init model_dict = util.transfer(self.model, torch.load(model_path)) File "/root/miniconda3/envs/ootd/lib/python3.10/site-packages/torch/serialization.py", line 815, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "/root/miniconda3/envs/ootd/lib/python3.10/site-packages/torch/serialization.py", line 1033, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: invalid load key, '\x05'.

xiyuxiaoxiao avatar Feb 23 '24 10:02 xiyuxiaoxiao

@xiyuxiaoxiao Hello, I am wondering if you fixed the bug.

LuckyOne09 avatar Feb 28 '24 11:02 LuckyOne09

no

xiyuxiaoxiao avatar Mar 01 '24 03:03 xiyuxiaoxiao

it's working fine for me - use python 3.11 you did install the requirements.txt ? here's my pip freeze. https://gist.github.com/johndpope/2132ea6acc195ad03b77e5ec61002b2d

Screenshot from 2024-04-15 15-08-33

python run_ootd.py --model_path ./examples/model/01008_00.jpg --cloth_path ./examples/garment/00055_00.jpg --scale 2.0 --sample 4 

johndpope avatar Apr 15 '24 05:04 johndpope