ghtning automatically upgraded your loaded checkpoint from v1.8.1 to v2.2.1. To apply the upgrade to your files permanently, run python -m pytorch_lightning.utilities.upgrade_checkpoint custom_nodes/ComfyUI-MotionDiff/ckpts/epoch=35-step=1000000.ckpt
!!! Exception during processing !!!
Traceback (most recent call last):
File "/mnt/c/shared/projects/ComfyUI/execution.py", line 151, in recursive_execute
output_data, output_ui = get_output_data(obj, input_data_all)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/mnt/c/shared/projects/ComfyUI/execution.py", line 81, in get_output_data
return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/mnt/c/shared/projects/ComfyUI/execution.py", line 74, in map_node_over_list
results.append(getattr(obj, func)(slice_dict(input_data_all, i)))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/mnt/c/shared/projects/ComfyUI/custom_nodes/ComfyUI-MotionDiff/nodes/human4d_nodes.py", line 49, in load
model, model_cfg = load_hmr2(DEFAULT_CHECKPOINT)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/mnt/c/shared/projects/ComfyUI/custom_nodes/ComfyUI-MotionDiff/motiondiff_modules/hmr2/models/init.py", line 89, in load_hmr2
model = HMR2.load_from_checkpoint(checkpoint_path, strict=False, cfg=model_cfg)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/pytorch_lightning/utilities/model_helpers.py", line 125, in wrapper
return self.method(cls, args, kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/pytorch_lightning/core/module.py", line 1581, in load_from_checkpoint
loaded = _load_from_checkpoint(
^^^^^^^^^^^^^^^^^^^^^^
File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/pytorch_lightning/core/saving.py", line 91, in _load_from_checkpoint
model = _load_state(cls, checkpoint, strict=strict, kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/pytorch_lightning/core/saving.py", line 158, in _load_state
obj = cls(_cls_kwargs)
^^^^^^^^^^^^^^^^^^
File "/mnt/c/shared/projects/ComfyUI/custom_nodes/ComfyUI-MotionDiff/motiondiff_modules/hmr2/models/hmr2_arch.py", line 52, in init
self.smpl = SMPL(smpl_cfg)
^^^^^^^^^^^^^^^^
File "/mnt/c/shared/projects/ComfyUI/custom_nodes/ComfyUI-MotionDiff/motiondiff_modules/hmr2/models/smpl_wrapper.py", line 18, in init
super(SMPL, self).init(args, kwargs)
File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/smplx/body_models.py", line 400, in init
super(SMPLLayer, self).init(
File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/smplx/body_models.py", line 137, in init
data_struct = Struct(**pickle.load(smpl_file,
^^^^^^^^^^^^^^^^^^^^^^
_pickle.UnpicklingError: pickle data was truncated
ghtning automatically upgraded your loaded checkpoint from v1.8.1 to v2.2.1. To apply the upgrade to your files permanently, run
python -m pytorch_lightning.utilities.upgrade_checkpoint custom_nodes/ComfyUI-MotionDiff/ckpts/epoch=35-step=1000000.ckpt!!! Exception during processing !!! Traceback (most recent call last): File "/mnt/c/shared/projects/ComfyUI/execution.py", line 151, in recursive_execute output_data, output_ui = get_output_data(obj, input_data_all) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/mnt/c/shared/projects/ComfyUI/execution.py", line 81, in get_output_data return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/mnt/c/shared/projects/ComfyUI/execution.py", line 74, in map_node_over_list results.append(getattr(obj, func)(slice_dict(input_data_all, i))) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/mnt/c/shared/projects/ComfyUI/custom_nodes/ComfyUI-MotionDiff/nodes/human4d_nodes.py", line 49, in load model, model_cfg = load_hmr2(DEFAULT_CHECKPOINT) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/mnt/c/shared/projects/ComfyUI/custom_nodes/ComfyUI-MotionDiff/motiondiff_modules/hmr2/models/init.py", line 89, in load_hmr2 model = HMR2.load_from_checkpoint(checkpoint_path, strict=False, cfg=model_cfg) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/pytorch_lightning/utilities/model_helpers.py", line 125, in wrapper return self.method(cls, args, kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/pytorch_lightning/core/module.py", line 1581, in load_from_checkpoint loaded = _load_from_checkpoint( ^^^^^^^^^^^^^^^^^^^^^^ File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/pytorch_lightning/core/saving.py", line 91, in _load_from_checkpoint model = _load_state(cls, checkpoint, strict=strict, kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/pytorch_lightning/core/saving.py", line 158, in _load_state obj = cls(_cls_kwargs) ^^^^^^^^^^^^^^^^^^ File "/mnt/c/shared/projects/ComfyUI/custom_nodes/ComfyUI-MotionDiff/motiondiff_modules/hmr2/models/hmr2_arch.py", line 52, in init self.smpl = SMPL(smpl_cfg) ^^^^^^^^^^^^^^^^ File "/mnt/c/shared/projects/ComfyUI/custom_nodes/ComfyUI-MotionDiff/motiondiff_modules/hmr2/models/smpl_wrapper.py", line 18, in init super(SMPL, self).init(args, kwargs) File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/smplx/body_models.py", line 400, in init super(SMPLLayer, self).init( File "/home/xlei/miniconda3/envs/py311/lib/python3.11/site-packages/smplx/body_models.py", line 137, in init data_struct = Struct(**pickle.load(smpl_file, ^^^^^^^^^^^^^^^^^^^^^^ _pickle.UnpicklingError: pickle data was truncatedthe error persists after I re-download the model.