Skip to content

Commit 8fea23c

Browse files
guangy10facebook-github-bot
authored andcommitted
version bump for mobilebert (#592)
Summary: v3.34.0 include the patch that fixed the mobilebert export Differential Revision: D49874303
1 parent fdaf6fa commit 8fea23c

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

.ci/docker/requirements-ci.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ sympy==1.12
66
timm==0.6.13
77
tomli==2.0.1
88
torchsr==1.0.4
9-
transformers==4.31.0
9+
transformers==4.34.0
1010
zstd==1.5.5.1
1111
pytest==7.2.0
1212
pytest-cov==4.1.0

examples/models/mobilebert/model.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,16 @@
55
# LICENSE file in the root directory of this source tree.
66

77
import logging
8+
import sys
89

910
import torch
1011

1112
from transformers import AutoTokenizer, MobileBertModel # @manual
1213

1314
from ..model_base import EagerModelBase
1415

16+
sys.setrecursionlimit(30000)
17+
1518

1619
class MobileBertModelExample(EagerModelBase):
1720
def __init__(self):

install_requirements.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,8 @@ pip install --force-reinstall --pre torchaudio=="${TORCH_AUDIO_VERSION}" -i http
3131
TIMM_VERSION=0.6.13
3232
pip install --pre timm==${TIMM_VERSION}
3333

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}
3636

3737
TORCHSR_VERSION=1.0.4
3838
pip install --pre torchsr==${TORCHSR_VERSION}

0 commit comments

Comments
 (0)