Skip to content
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

[Crash] Crash while loading AlibabaNLP/gte-base ONNX model #21322

Open
vivekg-ai opened this issue Jul 11, 2024 · 5 comments
Open

[Crash] Crash while loading AlibabaNLP/gte-base ONNX model #21322

vivekg-ai opened this issue Jul 11, 2024 · 5 comments
Labels
model:transformer issues related to a transformer model: BERT, GPT2, Hugging Face, Longformer, T5, etc. platform:mobile issues related to ONNX Runtime mobile; typically submitted using template

Comments

@vivekg-ai
Copy link

vivekg-ai commented Jul 11, 2024

Describe the issue

We are seeing a crash report while loading certain models (e.g. AlibabaNLP/gte-base - : https://huggingface.co/Alibaba-NLP/gte-base-en-v1.5)

The crash is seen on macOS, and here is the crash location -

image

To reproduce

Try loading the given model from the URL - https://huggingface.co/Alibaba-NLP/gte-base-en-v1.5

See the crash while inferencing.

Urgency

No response

Platform

Mac

OS Version

14.5

ONNX Runtime Installation

Released Package

ONNX Runtime Version or Commit ID

https://github.com/thebrowsercompany/onnxruntime-swift-package-manager

ONNX Runtime API

Obj-C

Architecture

ARM64

Execution Provider

Default CPU

Execution Provider Library Version

No response

@github-actions github-actions bot added model:transformer issues related to a transformer model: BERT, GPT2, Hugging Face, Longformer, T5, etc. platform:mobile issues related to ONNX Runtime mobile; typically submitted using template labels Jul 11, 2024
@yufenglee
Copy link
Member

@vivekg-ai, could you please share a call stack and how is the model converted to ONNX?

@vivekg-ai
Copy link
Author

We are using the ONNX model directly available here - https://huggingface.co/Alibaba-NLP/gte-base-en-v1.5/tree/main/onnx

Unfortunately, the call stack doesn't have much since we are using the release build of the official SPM package of the ONNX Runtime from here - https://github.com/thebrowsercompany/onnxruntime-swift-package-manager

@tianleiwu
Copy link
Contributor

image
That only indicates that the model is failed to create session.

@vivekg-ai, There are multiple models in https://huggingface.co/Alibaba-NLP/gte-base-en-v1.5/tree/main/onnx. Which one did you use?

@vivekg-ai
Copy link
Author

We tried -
CleanShot 2024-07-12 at 00 10 36

@tianleiwu
Copy link
Contributor

I tried it in my windows machine. It is good to create session from python API

import onnxruntime
ort_session = onnxruntime.InferenceSession("model.onnx", providers=['CPUExecutionProvider'])

Could be related to Mac. @skottmckay, @edgchen1 to take a look at onnxruntime-swift-package-manager.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
model:transformer issues related to a transformer model: BERT, GPT2, Hugging Face, Longformer, T5, etc. platform:mobile issues related to ONNX Runtime mobile; typically submitted using template
Projects
None yet
Development

No branches or pull requests

3 participants