fix(granite): Add MIN_TORCH_VERSION for recently fixed torch bug

huggingface#2043 (comment) Branch: OnnxGranite Signed-off-by: Gabe Goodhart <[email protected]>
gabe-l-hart · Oct 22, 2024 · 186848a · 186848a
1 parent 5925174
commit 186848a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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):