From 186848a3382207cf1f0eb546ebe258ac24a71499 Mon Sep 17 00:00:00 2001 From: Gabe Goodhart Date: Tue, 22 Oct 2024 16:17:17 -0600 Subject: [PATCH] fix(granite): Add MIN_TORCH_VERSION for recently fixed torch bug https://github.com/huggingface/optimum/pull/2043#issuecomment-2427975461 Branch: OnnxGranite Signed-off-by: Gabe Goodhart --- optimum/exporters/onnx/model_configs.py | 1 + 1 file changed, 1 insertion(+) diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py index c5faffcf0b9..4f5e47d5df6 100644 --- a/optimum/exporters/onnx/model_configs.py +++ b/optimum/exporters/onnx/model_configs.py @@ -292,6 +292,7 @@ class GemmaOnnxConfig(LlamaOnnxConfig): class GraniteOnnxConfig(LlamaOnnxConfig): MIN_TRANSFORMERS_VERSION = version.parse("4.45.0") + MIN_TORCH_VERSION = version.parse("2.5.0") class PhiOnnxConfig(TextDecoderWithPositionIdsOnnxConfig):