convert-to-onnx / requirements.txt
Felladrin's picture
Revert to a previous version of transformers[torch] package, as the newer is breaking the conversions
962e56f verified
raw
history blame contribute delete
213 Bytes
huggingface_hub
streamlit
transformers[torch]==4.49.0
onnxruntime==1.20.1
optimum@git+https://github.com/huggingface/optimum.git@b04feaea78cda58d79b8da67dca3fd0c4ab33435
onnx==1.17.0
tqdm==4.67.1
onnxslim==0.1.48