[Bug] Optimization step for unet fails after 'Protobuf parsing failed' about olive HOT 10 CLOSED

what's your protobuf version?

from olive.

BTW, could you please check whether the model file is valid or not? could it be open by https://netron.app/?

from olive.

@guotuofeng

Protobuf version : protobuf==3.20.3

When I tried to load the same model file on the website you linked it failed as well with: Error loading ONNX model. File format is not onnx.ModelProto (Array buffer allocation failed).

Does this mean the file is corrupted?

Rest of pip freeze from the conda environment:

accelerate==0.26.1
aiohttp==3.9.3
aiosignal==1.3.1
alembic==1.13.1
annotated-types==0.6.0
attrs==23.2.0
certifi==2023.11.17
charset-normalizer==3.3.2
colorama==0.4.6
coloredlogs==15.0.1
colorlog==6.8.2
datasets==2.16.1
diffusers==0.26.0
dill==0.3.7
filelock==3.13.1
flatbuffers==23.5.26
frozenlist==1.4.1
fsspec==2023.10.0
greenlet==3.0.3
huggingface-hub==0.20.3
humanfriendly==10.0
idna==3.6
importlib-metadata==7.0.1
invisible-watermark==0.2.0
Jinja2==3.1.3
lightning-utilities==0.10.1
Mako==1.3.2
MarkupSafe==2.1.4
mpmath==1.3.0
multidict==6.0.5
multiprocess==0.70.15
networkx==3.2.1
numpy==1.26.3
olive-ai @ file:///C:/Users/malhamdan/thisonewillwork/Olive
onnx==1.15.0
onnxruntime-directml==1.17.0
opencv-python==4.9.0.80
optimum==1.16.2
optuna==3.5.0
packaging==23.2
pandas==2.2.0
pillow==10.2.0
protobuf==3.20.3
psutil==5.9.8
pyarrow==15.0.0
pyarrow-hotfix==0.6
pydantic==2.6.0
pydantic_core==2.16.1
pyreadline3==3.4.1
python-dateutil==2.8.2
pytz==2023.4
PyWavelets==1.5.0
PyYAML==6.0.1
regex==2023.12.25
requests==2.31.0
safetensors==0.4.2
sentencepiece==0.1.99
six==1.16.0
SQLAlchemy==2.0.25
sympy==1.12
tabulate==0.9.0
tokenizers==0.15.1
torch==2.2.0
torchmetrics==1.3.0.post0
tqdm==4.66.1
transformers==4.37.2
typing_extensions==4.9.0
tzdata==2023.4
urllib3==2.2.0
xxhash==3.4.1
yarl==1.9.4
zipp==3.17.0

from olive.

I tried reinstalling https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0 directly locally using huggingface-cli to make sure there is no corruption, and I then modified the stable_diffusion_xl.py to point to my local model, but I am still getting the same protobuf error.

from olive.

Does this mean the file is corrupted?

If the model file cannot be loaded by the netron app, it means the model file is corrupted.

from olive.

Does this mean the file is corrupted?

If the model file cannot be loaded by the netron app, it means the model file is corrupted.

I download the sdxl multiple times using huggingface_cli, and also using the automatic download provided by the Olive python code to ensure there is no prior corruption, and it all leads to the same error. This means Olive is not transforming the model to a proper .onnx model correctly? What can be done?

Thank you.

from olive.

it seems similar with microsoft/onnxruntime#10892? what's the model size?

from olive.

I faced the same issue, how to solve it

from olive.

I faced the same issue, how to solve it

what's your model size?

from olive.

PR #1069 should fix the issue that was specific to python 3.11 when saving large models.

Please reopen the issue if the problem persists.

from olive.