-
Notifications
You must be signed in to change notification settings - Fork 839
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
INVALID_PROTOBUF #189
Comments
Please download the checkpoints from https://huggingface.co/levihsu/OOTDiffusion/tree/main/checkpoints |
2024-06-23 09:11:11.163291: E external/local_xla/xla/stream_executor/cuda/cuda_dnn.cc:9261] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered I am trying this in colab with GPU and facing above issue, Also tried to replace the parsing files from the below location and still the same issue, Could you please guide further. Thanks https://huggingface.co/levihsu/OOTDiffusion/tree/main/checkpoints |
Traceback (most recent call last):
File "/home//Downloads/cloth_try_on/OOTDiffusion/run/run_ootd.py", line 30, in
parsing_model = Parsing(args.gpu_id)
File "/home//Downloads/cloth_try_on/OOTDiffusion/preprocess/humanparsing/run_parsing.py", line 20, in init
self.session = ort.InferenceSession(os.path.join(Path(file).absolute().parents[2].absolute(), 'checkpoints/humanparsing/parsing_atr.onnx'),
File "/home//anaconda3/envs/ootd/lib/python3.10/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 419, in init
self._create_inference_session(providers, provider_options, disabled_optimizers)
File "/home//anaconda3/envs/ootd/lib/python3.10/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 452, in _create_inference_session
sess = C.InferenceSession(session_options, self._model_path, True, self._read_config_from_model)
onnxruntime.capi.onnxruntime_pybind11_state.InvalidProtobuf: [ONNXRuntimeError] : 7 : INVALID_PROTOBUF : Load model from /home/****/Downloads/cloth_try_on/OOTDiffusion/checkpoints/humanparsing/parsing_atr.onnx failed:Protobuf parsing failed.
Could you pls help to resolve this issue?
Thanks
The text was updated successfully, but these errors were encountered: