mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-03 03:31:05 +06:00
Pin minimum PyTorch version for BLOOM ONNX export (#19046)
This commit is contained in:
parent
0a42b61ede
commit
9b80a0bc18
@ -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,
|
||||
|
Loading…
Reference in New Issue
Block a user