File tree 3 files changed +6
-23
lines changed
src/deepsparse/transformers
3 files changed +6
-23
lines changed Original file line number Diff line number Diff line change @@ -145,9 +145,8 @@ def _parse_requirements_file(file_path):
145
145
]
146
146
_yolov8_integration_deps = _yolo_integration_deps + ["ultralytics==8.0.124" ]
147
147
_transformers_integration_deps = [
148
- f"{ 'nm-transformers' if is_release else 'nm-transformers-nightly' } "
149
- f"~={ version_base } " ,
150
- "datasets<=2.11" ,
148
+ "transformers<4.35" ,
149
+ "datasets<2.13" ,
151
150
"scikit-learn" ,
152
151
"seqeval" ,
153
152
]
@@ -166,7 +165,7 @@ def _parse_requirements_file(file_path):
166
165
_clip_deps = [
167
166
"open_clip_torch==2.20.0" ,
168
167
"scipy<1.9.2,>=1.8" ,
169
- f" { 'nm- transformers' if is_release else 'nm-transformers-nightly' } " ,
168
+ " transformers<4.35 " ,
170
169
]
171
170
172
171
_torch_deps = ["torch>=1.7.0,<=2.0" ]
Original file line number Diff line number Diff line change 19
19
20
20
# flake8: noqa
21
21
22
- import logging as _logging
23
22
24
23
from deepsparse .analytics import deepsparse_analytics as _analytics
25
24
35
34
raise ImportError ("Please install deepsparse[transformers] to use this pathway" )
36
35
37
36
38
- _LOGGER = _logging .getLogger (__name__ )
39
-
40
-
41
- def _check_transformers_install ():
42
- import transformers as _transformers
43
-
44
- if not getattr (_transformers , "NM_INTEGRATED" , False ):
45
- _LOGGER .warning (
46
- "The neuralmagic fork of transformers may not be installed. It can be "
47
- "installed via `pip install nm_transformers`"
48
- )
49
-
50
-
51
- _check_transformers_install ()
52
-
53
-
54
37
from .helpers import *
55
38
from .loaders import *
56
39
from .pipelines import *
Original file line number Diff line number Diff line change @@ -26,8 +26,9 @@ def install_reqs():
26
26
"-m" ,
27
27
"pip" ,
28
28
"install" ,
29
- "-r" ,
30
- "examples/twitter-nlp/requirements.txt" ,
29
+ "git+https://github.com/twintproject/"
30
+ "twint@e7c8a0c764f6879188e5c21e25fb6f1f856a7221#egg=twint" ,
31
+ "rich>=12.2.0" ,
31
32
]
32
33
)
33
34
You can’t perform that action at this time.
0 commit comments