Skip to content

After ComfyUI Custom Nodes Manager Update, Error Loading DINOv2 Model #67

@SuperWeapons2770

Description

@SuperWeapons2770

I had this working a while ago, maybe December or early January, and I decided that I would see if the comfyui custom node manager update would just allow this to update nicely. It appears it does not.

When I updated, ComfyUI no longer let this run as multiple fields in the IF_TrellisImageTo3D Node and the IF_TrellisCheckpointLoader Node. I set values to those that were acceptable, and ComfyUI attempts to execute, and appears to be successful in loading the checkpoint up until it attempts to load whatever DINOv2 is. I think this may be a new model that is needed?

ComfyUI says:

IF_TrellisCheckpointLoader
Error loading DINOv2 model: PytorchStreamReader failed reading zip archive: failed finding central directory

Is this a new model that needs to be downloaded? I don't think that was in this by default.

Thanks,

Python dump:

got prompt
Setting attention backend to: flash_attn
Setting sparse backend to: spconv
Set spconv algorithm to: native
Loading local model: ss_dec_conv3d_16l8_fp16
Loading local model: ss_flow_img_dit_L_16l8_fp16
Loading local model: slat_dec_gs_swin8_B_64l8gs32_fp16
Loading local model: slat_dec_rf_swin8_B_64l8r16_fp16
Loading local model: slat_dec_mesh_swin8_B_64l8m256c_fp16
Loading local model: slat_flow_img_dit_L_64l8p2_fp16
Loading DINOv2 model from C:\AI\ComfyUI_micromamba\ComfyUI\models\classifiers\dinov2_vitl14_reg.pth
Using cache found in C:\Users\You Don't get to know my username/.cache\torch\hub\facebookresearch_dinov2_main
using MLP layer as FFN
Error loading DINOv2 model: Error loading DINOv2 model: PytorchStreamReader failed reading zip archive: failed finding central directory
Error loading TRELLIS model: Error loading DINOv2 model: PytorchStreamReader failed reading zip archive: failed finding central directory
!!! Exception during processing !!! Error loading DINOv2 model: PytorchStreamReader failed reading zip archive: failed finding central directory
Traceback (most recent call last):
File "C:\AI\ComfyUI_micromamba\ComfyUI\custom_nodes\ComfyUI-IF_Trellis\trellis_model_manager.py", line 218, in load_dinov2
model.load_state_dict(torch.load(model_path))
^^^^^^^^^^^^^^^^^^^^^^
File "C:\AI\micromamba\envs\3d12\Lib\site-packages\torch\serialization.py", line 1326, in load
with _open_zipfile_reader(opened_file) as opened_zipfile:
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\AI\micromamba\envs\3d12\Lib\site-packages\torch\serialization.py", line 671, in init
super().init(torch._C.PyTorchFileReader(name_or_buffer))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "C:\AI\ComfyUI_micromamba\ComfyUI\execution.py", line 328, in execute
output_data, output_ui, has_subgraph = get_output_data(obj, input_data_all, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\AI\ComfyUI_micromamba\ComfyUI\execution.py", line 203, in get_output_data
return_values = _map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\AI\ComfyUI_micromamba\ComfyUI\execution.py", line 174, in _map_node_over_list
process_inputs(input_dict, i)
File "C:\AI\ComfyUI_micromamba\ComfyUI\execution.py", line 163, in process_inputs
results.append(getattr(obj, func)(**inputs))
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\AI\ComfyUI_micromamba\ComfyUI\custom_nodes\ComfyUI-IF_Trellis\IF_TrellisCheckpointLoader.py", line 158, in load_model
pipeline = TrellisImageTo3DPipeline.from_pretrained(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\AI\ComfyUI_micromamba\ComfyUI\custom_nodes\ComfyUI-IF_Trellis\trellis\pipelines\trellis_image_to_3d.py", line 87, in from_pretrained
new_pipeline._init_image_cond_model(model_name)
File "C:\AI\ComfyUI_micromamba\ComfyUI\custom_nodes\ComfyUI-IF_Trellis\trellis\pipelines\trellis_image_to_3d.py", line 104, in _init_image_cond_model
dinov2_model = model_manager.load_dinov2(name)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\AI\ComfyUI_micromamba\ComfyUI\custom_nodes\ComfyUI-IF_Trellis\trellis_model_manager.py", line 233, in load_dinov2
raise RuntimeError(f"Error loading DINOv2 model: {str(e)}")
RuntimeError: Error loading DINOv2 model: PytorchStreamReader failed reading zip archive: failed finding central directory

Prompt executed in 13.95 seconds
fatal: detected dubious ownership in repository at 'C:/AI/ComfyUI_micromamba/ComfyUI'
'C:/AI/ComfyUI_micromamba/ComfyUI' is owned by:
'S-1-5-32-544'
but the current user is:
'S-1-5-21-2296672686-2570481813-1124366436-1001'
To add an exception for this directory, call:

    git config --global --add safe.directory C:/AI/ComfyUI_micromamba/ComfyUI

Failed to get ComfyUI version: Command '['git', 'describe', '--tags']' returned non-zero exit status 128.

Examining this dump seems to indicate it can't get the correct path on line 213 of C:\AI\ComfyUI_micromamba\ComfyUI\custom_nodes\ComfyUI-IF_Trellis\trellis_model_manager.py but I confirmed that C:\AI\ComfyUI_micromamba\ComfyUI\models\classifiers\dinov2_vitl14_reg.pth is in the path at the specified place.

Thanks

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions