[ERROR] !!! Exception during processing !!! Error while deserializing header: InvalidHeaderDeserialization

[Hazukiaoi]

When I use the Pixal3D workflow, the Trellis2 - Sparse Generator node throws an execution error. I have verified that the files in the model path are correct.

And the error log:

[INFO] [ComfyUI-Manager] All startup tasks have been completed.
[06/03 22:39:02][INFO] [ComfyUI-Manager] All startup tasks have been completed.
[06/03 22:39:02][INFO] [ComfyUI-Manager] All startup tasks have been completed.
[INFO] got prompt
[06/03 22:57:22][INFO] got prompt
[06/03 22:57:22][INFO] got prompt
mmgp not installed
Loading Image Cond model ...
Loading weights: 100%|███████████████████████████████████████████| 415/415 [00:01<00:00, 340.10it/s]
Loading MoGe model ...
[INFO] using MLP layer as FFN
[06/03 22:57:26][INFO] using MLP layer as FFN
[06/03 22:57:26][INFO] using MLP layer as FFN
{'camera_angle_x': 0.5493824847775403, 'distance': 1.7742115259170532, 'mesh_scale': 1.0}
Loading Pixal3D Image Cond SS Model ...
Loading weights: 100%|███████████████████████████████████████████| 415/415 [00:01<00:00, 338.12it/s]
Getting Proj Image Cond ...
Loading Sparse Structure model ...
[ERROR] !!! Exception during processing !!! Error while deserializing header: InvalidHeaderDeserialization
[06/03 22:57:37][ERROR] !!! Exception during processing !!! Error while deserializing header: InvalidHeaderDeserialization
[06/03 22:57:37][ERROR] !!! Exception during processing !!! Error while deserializing header: InvalidHeaderDeserialization
[ERROR] Traceback (most recent call last):
  File "/mnt/AI/ComfyUI/execution.py", line 536, in execute
    output_data, output_ui, has_subgraph, has_pending_tasks = await get_output_data(prompt_id, unique_id, obj, input_data_all, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
  File "/mnt/AI/ComfyUI/execution.py", line 336, in get_output_data
    return_values = await _async_map_node_over_list(prompt_id, unique_id, obj, input_data_all, obj.FUNCTION, allow_interrupt=True, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
  File "/mnt/AI/ComfyUI/execution.py", line 310, in _async_map_node_over_list
    await process_inputs(input_dict, i)
  File "/mnt/AI/ComfyUI/execution.py", line 298, in process_inputs
    result = f(**inputs)
  File "/mnt/AI/ComfyUI/custom_nodes/ComfyUI-Trellis2/nodes.py", line 4003, in process
    pipeline.load_sparse_structure_model()
  File "/mnt/AI/ComfyUI/custom_nodes/ComfyUI-Trellis2/trellis2/pipelines/trellis2_image_to_3d.py", line 353, in load_sparse_structure_model
    self.models['sparse_structure_flow_model'] = models.from_pretrained(f"{self.path}/{self._pretrained_args['models']['sparse_structure_flow_model']}")
  File "/mnt/AI/ComfyUI/custom_nodes/ComfyUI-Trellis2/trellis2/models/__init__.py", line 69, in from_pretrained
    model.load_state_dict(load_file(model_file), strict=False)
  File "/home/cuda12/anaconda3/envs/comfyui/lib/python3.10/site-packages/safetensors/torch.py", line 313, in load_file
    with safe_open(filename, framework="pt", device=device) as f:
safetensors_rust.SafetensorError: Error while deserializing header: InvalidHeaderDeserialization

[06/03 22:57:37][ERROR] Traceback (most recent call last):
  File "/mnt/AI/ComfyUI/execution.py", line 536, in execute
    output_data, output_ui, has_subgraph, has_pending_tasks = await get_output_data(prompt_id, unique_id, obj, input_data_all, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
  File "/mnt/AI/ComfyUI/execution.py", line 336, in get_output_data
    return_values = await _async_map_node_over_list(prompt_id, unique_id, obj, input_data_all, obj.FUNCTION, allow_interrupt=True, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
  File "/mnt/AI/ComfyUI/execution.py", line 310, in _async_map_node_over_list
    await process_inputs(input_dict, i)
  File "/mnt/AI/ComfyUI/execution.py", line 298, in process_inputs
    result = f(**inputs)
  File "/mnt/AI/ComfyUI/custom_nodes/ComfyUI-Trellis2/nodes.py", line 4003, in process
    pipeline.load_sparse_structure_model()
  File "/mnt/AI/ComfyUI/custom_nodes/ComfyUI-Trellis2/trellis2/pipelines/trellis2_image_to_3d.py", line 353, in load_sparse_structure_model
    self.models['sparse_structure_flow_model'] = models.from_pretrained(f"{self.path}/{self._pretrained_args['models']['sparse_structure_flow_model']}")
  File "/mnt/AI/ComfyUI/custom_nodes/ComfyUI-Trellis2/trellis2/models/__init__.py", line 69, in from_pretrained
    model.load_state_dict(load_file(model_file), strict=False)
  File "/home/cuda12/anaconda3/envs/comfyui/lib/python3.10/site-packages/safetensors/torch.py", line 313, in load_file
    with safe_open(filename, framework="pt", device=device) as f:
safetensors_rust.SafetensorError: Error while deserializing header: InvalidHeaderDeserialization

[06/03 22:57:37][ERROR] Traceback (most recent call last):
  File "/mnt/AI/ComfyUI/execution.py", line 536, in execute
    output_data, output_ui, has_subgraph, has_pending_tasks = await get_output_data(prompt_id, unique_id, obj, input_data_all, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
  File "/mnt/AI/ComfyUI/execution.py", line 336, in get_output_data
    return_values = await _async_map_node_over_list(prompt_id, unique_id, obj, input_data_all, obj.FUNCTION, allow_interrupt=True, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
  File "/mnt/AI/ComfyUI/execution.py", line 310, in _async_map_node_over_list
    await process_inputs(input_dict, i)
  File "/mnt/AI/ComfyUI/execution.py", line 298, in process_inputs
    result = f(**inputs)
  File "/mnt/AI/ComfyUI/custom_nodes/ComfyUI-Trellis2/nodes.py", line 4003, in process
    pipeline.load_sparse_structure_model()
  File "/mnt/AI/ComfyUI/custom_nodes/ComfyUI-Trellis2/trellis2/pipelines/trellis2_image_to_3d.py", line 353, in load_sparse_structure_model
    self.models['sparse_structure_flow_model'] = models.from_pretrained(f"{self.path}/{self._pretrained_args['models']['sparse_structure_flow_model']}")
  File "/mnt/AI/ComfyUI/custom_nodes/ComfyUI-Trellis2/trellis2/models/__init__.py", line 69, in from_pretrained
    model.load_state_dict(load_file(model_file), strict=False)
  File "/home/cuda12/anaconda3/envs/comfyui/lib/python3.10/site-packages/safetensors/torch.py", line 313, in load_file
    with safe_open(filename, framework="pt", device=device) as f:
safetensors_rust.SafetensorError: Error while deserializing header: InvalidHeaderDeserialization

[entropy010]

Hi,
I had the same issue and managed to fix it.
Your safetensors module is too old.

pip uninstall safetensors
pip install -U "safetensors>=0.7.0"

I did pip install safetensors to version 0.8.0 ... now it works.
If you have a followup issue upon starting ComfyUI with _safetensors_rust.pyd* (something like that) then check the folder in \Lib\site-packages\safetensors\ and delete every _safetensors_rust_cp312*.pyd and reinstall safetensors again.

Hope that helps