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 25, 2024 · fbd051a · fbd051a
1 parent 05c301e
commit fbd051a
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):