-
Notifications
You must be signed in to change notification settings - Fork 138
Description
Loads checkpoint by local backend from path: weights/dwpose/dw-ll_ucoco_384.pth
Traceback (most recent call last):
File "/mnt/d/three/gatherModels/video_chat/app.py", line 48, in
from src.pipeline_llm import llm_pipeline
File "/mnt/d/three/gatherModels/video_chat/src/pipeline_llm.py", line 25, in
from src.thg import Muse_Talk
File "/mnt/d/three/gatherModels/video_chat/src/thg.py", line 24, in
from src.musetalk.utils.preprocessing import get_landmark_and_bbox,read_imgs,coord_placeholder
File "/mnt/d/three/gatherModels/video_chat/src/musetalk/utils/preprocessing.py", line 19, in
model = init_model(config_file, checkpoint_file, device=device)
File "/root/anaconda3/envs/metahuman310/lib/python3.10/site-packages/mmpose/apis/inference.py", line 110, in init_model
ckpt = load_checkpoint(model, checkpoint, map_location='cpu')
File "/root/anaconda3/envs/metahuman310/lib/python3.10/site-packages/mmengine/runner/checkpoint.py", line 636, in load_checkpoint
checkpoint = _load_checkpoint(filename, map_location, logger)
File "/root/anaconda3/envs/metahuman310/lib/python3.10/site-packages/mmengine/runner/checkpoint.py", line 548, in _load_checkpoint
return CheckpointLoader.load_checkpoint(filename, map_location, logger)
File "/root/anaconda3/envs/metahuman310/lib/python3.10/site-packages/mmengine/runner/checkpoint.py", line 330, in load_checkpoint
return checkpoint_loader(filename, map_location)
File "/root/anaconda3/envs/metahuman310/lib/python3.10/site-packages/mmengine/runner/checkpoint.py", line 347, in load_from_local
checkpoint = torch.load(filename, map_location=map_location)
File "/root/anaconda3/envs/metahuman310/lib/python3.10/site-packages/torch/serialization.py", line 1548, in load
raise pickle.UnpicklingError(_get_wo_message(str(e))) from None
_pickle.UnpicklingError: Weights only load failed. In PyTorch 2.6, we changed the default value of the weights_only
argument in torch.load
from False
to True
. Re-running torch.load
with weights_only
set to False
will likely succeed, but it can result in arbitrary code execution. Do it only if you got the file from a trusted source.
Please file an issue with the following so that we can make weights_only=True
compatible with your use case: WeightsUnpickler error: Unsupported operand 118
Check the documentation of torch.load to learn more about types accepted by default with weights_only https://pytorch.org/docs/stable/generated/torch.load.html.