Unable to generate ONNX graph for GPT-2. #16
Labels
bug
Something isn't working
documentation
Improvements or additions to documentation
help wanted
Extra attention is needed
Hello, I'm Jongmin Kim, a student studying DNN accelerator, and I have been interested in your simulator.
According to the README, the following command should generate an ONNX graph for GPT-2:
python3 ./scripts/generate_transformer_onnx.py --model gpt2
However, it results in the following error message:
I have verified that the version is 22, and I don’t believe this is the issue, as the latest versions of
onnx
andonnxruntime
are installed, as specified in the Dockerfile.I am currently attempting to export ONNX graphs for GPT-2, Llama 3, and OPT using the ONNX optimizer (
AutoModelForXXX
,ORTModelForXXX
, etc.).However, I’m wondering if there are any simpler or more efficient methods to resolve this issue.
Thank you for your time, and I look forward to your guidance.
The text was updated successfully, but these errors were encountered: