Open florestefano1975 opened 5 months ago
2024-03-26T17:02:18.884730049Z 2024-03-26 17:02:18.884 | ERROR | ComfyUI-BiRefNet.BiRefNet_node:load:65 - Failed to load the model: invalid load key, 'v'. 2024-03-26T17:02:18.890070547Z !!! Exception during processing !!! 2024-03-26T17:02:18.895161496Z Traceback (most recent call last): 2024-03-26T17:02:18.895166696Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/BiRefNet_node.py", line 49, in load 2024-03-26T17:02:18.895168566Z self.model = BiRefNet() 2024-03-26T17:02:18.895170136Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/models/baseline.py", line 28, in init 2024-03-26T17:02:18.895171836Z self.bb = build_backbone(self.config.bb, pretrained=True) 2024-03-26T17:02:18.895173186Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/models/backbones/build_backbone.py", line 25, in build_backbone 2024-03-26T17:02:18.895174406Z bb = load_weights(bb, bb_name) 2024-03-26T17:02:18.895175526Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/models/backbones/build_backbone.py", line 29, in load_weights 2024-03-26T17:02:18.895176716Z save_model = torch.load(config.weights[model_name], map_location=torch.device('cpu')) 2024-03-26T17:02:18.895177946Z File "/venv/lib/python3.10/site-packages/torch/serialization.py", line 815, in load 2024-03-26T17:02:18.895179146Z return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args) 2024-03-26T17:02:18.895180276Z File "/venv/lib/python3.10/site-packages/torch/serialization.py", line 1033, in _legacy_load 2024-03-26T17:02:18.895181406Z magic_number = pickle_module.load(f, pickle_load_args) 2024-03-26T17:02:18.895182536Z _pickle.UnpicklingError: invalid load key, 'v'. 2024-03-26T17:02:18.895183646Z 2024-03-26T17:02:18.895184746Z During handling of the above exception, another exception occurred: 2024-03-26T17:02:18.895186376Z 2024-03-26T17:02:18.895187456Z Traceback (most recent call last): 2024-03-26T17:02:18.895188596Z File "/workspace/ComfyUI/execution.py", line 151, in recursive_execute 2024-03-26T17:02:18.895189716Z output_data, output_ui = get_output_data(obj, input_data_all) 2024-03-26T17:02:18.895190816Z File "/workspace/ComfyUI/execution.py", line 81, in get_output_data 2024-03-26T17:02:18.895192236Z return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True) 2024-03-26T17:02:18.895204716Z File "/workspace/ComfyUI/execution.py", line 74, in map_node_over_list 2024-03-26T17:02:18.895205966Z results.append(getattr(obj, func)(**slice_dict(input_data_all, i))) 2024-03-26T17:02:18.895207096Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/BiRefNet_node.py", line 101, in matting 2024-03-26T17:02:18.895208196Z self.load(weight_path, device=device) 2024-03-26T17:02:18.895209376Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/BiRefNet_node.py", line 67, in load 2024-03-26T17:02:18.895210466Z raise RuntimeError(f"Model loading failed: {e}") 2024-03-26T17:02:18.895211656Z RuntimeError: Model loading failed: invalid load key, 'v'.
base on your post log, I guess this is due to you did not download full of model checkpoitns, trouble shooting is following
du -sh *
to check the file size, compare the result with Huggingface dataset pageIf the file size matched, I guess sth happend due to checkpoint transferring or storing. the trough shooting is following
sha256sum *
and comapre the result with BiRefNet checkpoint sha256 on huggingface.BTW, the ultimate solution is redownload checkpoint from page based on your post info
I have downloaded everything 5 times, but I always get the same error. I will try again. Thank you.
2024-03-26T17:02:18.884730049Z 2024-03-26 17:02:18.884 | ERROR | ComfyUI-BiRefNet.BiRefNet_node:load:65 - Failed to load the model: invalid load key, 'v'. 2024-03-26T17:02:18.890070547Z !!! Exception during processing !!! 2024-03-26T17:02:18.895161496Z Traceback (most recent call last): 2024-03-26T17:02:18.895166696Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/BiRefNet_node.py", line 49, in load 2024-03-26T17:02:18.895168566Z self.model = BiRefNet() 2024-03-26T17:02:18.895170136Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/models/baseline.py", line 28, in init 2024-03-26T17:02:18.895171836Z self.bb = build_backbone(self.config.bb, pretrained=True) 2024-03-26T17:02:18.895173186Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/models/backbones/build_backbone.py", line 25, in build_backbone 2024-03-26T17:02:18.895174406Z bb = load_weights(bb, bb_name) 2024-03-26T17:02:18.895175526Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/models/backbones/build_backbone.py", line 29, in load_weights 2024-03-26T17:02:18.895176716Z save_model = torch.load(config.weights[model_name], map_location=torch.device('cpu')) 2024-03-26T17:02:18.895177946Z File "/venv/lib/python3.10/site-packages/torch/serialization.py", line 815, in load 2024-03-26T17:02:18.895179146Z return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args) 2024-03-26T17:02:18.895180276Z File "/venv/lib/python3.10/site-packages/torch/serialization.py", line 1033, in _legacy_load 2024-03-26T17:02:18.895181406Z magic_number = pickle_module.load(f, pickle_load_args) 2024-03-26T17:02:18.895182536Z _pickle.UnpicklingError: invalid load key, 'v'. 2024-03-26T17:02:18.895183646Z 2024-03-26T17:02:18.895184746Z During handling of the above exception, another exception occurred: 2024-03-26T17:02:18.895186376Z 2024-03-26T17:02:18.895187456Z Traceback (most recent call last): 2024-03-26T17:02:18.895188596Z File "/workspace/ComfyUI/execution.py", line 151, in recursive_execute 2024-03-26T17:02:18.895189716Z output_data, output_ui = get_output_data(obj, input_data_all) 2024-03-26T17:02:18.895190816Z File "/workspace/ComfyUI/execution.py", line 81, in get_output_data 2024-03-26T17:02:18.895192236Z return_values = map_node_over_list(obj, input_data_all, obj.FUNCTION, allow_interrupt=True) 2024-03-26T17:02:18.895204716Z File "/workspace/ComfyUI/execution.py", line 74, in map_node_over_list 2024-03-26T17:02:18.895205966Z results.append(getattr(obj, func)(**slice_dict(input_data_all, i))) 2024-03-26T17:02:18.895207096Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/BiRefNet_node.py", line 101, in matting 2024-03-26T17:02:18.895208196Z self.load(weight_path, device=device) 2024-03-26T17:02:18.895209376Z File "/workspace/ComfyUI/custom_nodes/ComfyUI-BiRefNet/BiRefNet_node.py", line 67, in load 2024-03-26T17:02:18.895210466Z raise RuntimeError(f"Model loading failed: {e}") 2024-03-26T17:02:18.895211656Z RuntimeError: Model loading failed: invalid load key, 'v'.