File tree Expand file tree Collapse file tree 3 files changed +6
-3
lines changed
examples/models/mobilebert Expand file tree Collapse file tree 3 files changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -6,7 +6,7 @@ sympy==1.12
6
6
timm==0.6.13
7
7
tomli==2.0.1
8
8
torchsr==1.0.4
9
- transformers==4.31 .0
9
+ transformers==4.34 .0
10
10
zstd==1.5.5.1
11
11
pytest==7.2.0
12
12
pytest-cov==4.1.0
Original file line number Diff line number Diff line change 5
5
# LICENSE file in the root directory of this source tree.
6
6
7
7
import logging
8
+ import sys
8
9
9
10
import torch
10
11
11
12
from transformers import AutoTokenizer , MobileBertModel # @manual
12
13
13
14
from ..model_base import EagerModelBase
14
15
16
+ sys .setrecursionlimit (30000 )
17
+
15
18
16
19
class MobileBertModelExample (EagerModelBase ):
17
20
def __init__ (self ):
Original file line number Diff line number Diff line change @@ -31,8 +31,8 @@ pip install --force-reinstall --pre torchaudio=="${TORCH_AUDIO_VERSION}" -i http
31
31
TIMM_VERSION=0.6.13
32
32
pip install --pre timm==${TIMM_VERSION}
33
33
34
- TRANSFORMERS_VERSION=4.31 .0
35
- pip install --pre transformers==${TRANSFORMERS_VERSION}
34
+ TRANSFORMERS_VERSION=4.34 .0
35
+ pip install --force-reinstall -- pre transformers==${TRANSFORMERS_VERSION}
36
36
37
37
TORCHSR_VERSION=1.0.4
38
38
pip install --pre torchsr==${TORCHSR_VERSION}
You can’t perform that action at this time.
0 commit comments