Skip to content

GitLab

"model/git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "d650ad398f7f1faaf2d4f71d820707b1a9467757"
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
Hide 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