Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "fcfd4ec7895a81b4ef67198a998c18e6c2e176bc"
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
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
src/transformers/models/bloom/configuration_bloom.py
View file @ 9b80a0bc
...@@ -16,6 +16,8 @@ ...@@ -16,6 +16,8 @@
from collections import OrderedDict from collections import OrderedDict
from typing import TYPE_CHECKING, Any, List, Mapping, Optional from typing import TYPE_CHECKING, Any, List, Mapping, Optional
from packaging import version
from transformers import is_torch_available from transformers import is_torch_available
...@@ -154,6 +156,9 @@ class BloomConfig(PretrainedConfig): ...@@ -154,6 +156,9 @@ class BloomConfig(PretrainedConfig):
class BloomOnnxConfig(OnnxConfigWithPast): class BloomOnnxConfig(OnnxConfigWithPast):
torch_onnx_minimum_version = version.parse("1.12")
def __init__( def __init__(
self, self,
config: PretrainedConfig, 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