Unverified Commit 9b80a0bc authored by lewtun's avatar lewtun Committed by GitHub
Browse files

Pin minimum PyTorch version for BLOOM ONNX export (#19046)

parent 0a42b61e
......@@ -16,6 +16,8 @@
from collections import OrderedDict
from typing import TYPE_CHECKING, Any, List, Mapping, Optional
from packaging import version
from transformers import is_torch_available
......@@ -154,6 +156,9 @@ class BloomConfig(PretrainedConfig):
class BloomOnnxConfig(OnnxConfigWithPast):
torch_onnx_minimum_version = version.parse("1.12")
def __init__(
self,
config: PretrainedConfig,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment