MusePose: a Pose-Driven Image-to-Video Framework for Virtual Human Generation
Other
2.29k
stars
167
forks
source link
Weights only load failed. Re-running `torch.load` with `weights_only` set to `False` will likely succeed, but it can result in arbitrary code execution. #72
the before command is all fine but the stage 2 show the model can't load
python test_stage_2.py --config ./configs/test_stage_2.yaml
Width: 768
Height: 768
Length: 300
Slice: 48
Overlap: 4
Classifier free guidance: 3.5
DDIM sampling steps : 20
skip 1
Some weights of the model checkpoint were not used when initializing UNet2DConditionModel:
['conv_norm_out.weight, conv_norm_out.bias, conv_out.weight, conv_out.bias']
/home/sysu/anaconda3/envs/manimate/lib/python3.8/site-packages/torch/_utils.py:776: UserWarning: TypedStorage is deprecated. It will be removed in the future and UntypedStorage will be the only storage class. This should only matter to you if you are using storages directly. To access UntypedStorage directly, use tensor.untyped_storage() instead of tensor.storage()
return self.fget.get(instance, owner)()
Traceback (most recent call last):
File "test_stage_2.py", line 237, in
main()
File "test_stage_2.py", line 87, in main
denoising_unet = UNet3DConditionModel.from_pretrained_2d(
File "/home/sysu/Desktop/animate/MusePose/musepose/models/unet_3d.py", line 638, in from_pretrained_2d
motion_state_dict = torch.load(
File "/home/sysu/anaconda3/envs/manimate/lib/python3.8/site-packages/torch/serialization.py", line 814, in load
raise pickle.UnpicklingError(UNSAFE_MESSAGE + str(e)) from None
_pickle.UnpicklingError: Weights only load failed. 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 get the file from a trusted source. WeightsUnpickler error: Unsupported operand 118
ailoov
commented
2 months ago
as the title and image shown, i met this error when running the command
python test_stage_2.py --config ./configs/test_stage_2.yamlthe before command is all fine but the stage 2 show the model can't load
python test_stage_2.py --config ./configs/test_stage_2.yaml Width: 768 Height: 768 Length: 300 Slice: 48 Overlap: 4 Classifier free guidance: 3.5 DDIM sampling steps : 20 skip 1 Some weights of the model checkpoint were not used when initializing UNet2DConditionModel: ['conv_norm_out.weight, conv_norm_out.bias, conv_out.weight, conv_out.bias'] /home/sysu/anaconda3/envs/manimate/lib/python3.8/site-packages/torch/_utils.py:776: UserWarning: TypedStorage is deprecated. It will be removed in the future and UntypedStorage will be the only storage class. This should only matter to you if you are using storages directly. To access UntypedStorage directly, use tensor.untyped_storage() instead of tensor.storage() return self.fget.get(instance, owner)() Traceback (most recent call last): File "test_stage_2.py", line 237, in
main()
File "test_stage_2.py", line 87, in main
denoising_unet = UNet3DConditionModel.from_pretrained_2d(
File "/home/sysu/Desktop/animate/MusePose/musepose/models/unet_3d.py", line 638, in from_pretrained_2d
motion_state_dict = torch.load(
File "/home/sysu/anaconda3/envs/manimate/lib/python3.8/site-packages/torch/serialization.py", line 814, in load
raise pickle.UnpicklingError(UNSAFE_MESSAGE + str(e)) from None
_pickle.UnpicklingError: Weights only load failed. Re-running
torch.loadwithweights_onlyset toFalsewill likely succeed, but it can result in arbitrary code execution.Do it only if you get the file from a trusted source. WeightsUnpickler error: Unsupported operand 118