Skip to content

Commit

Permalink
Remove explicit sentencepiece / protobuf deps (#900)
Browse files Browse the repository at this point in the history
* Remove explicit sentencepiece / protobuf deps as they optional deps for rarely used model configs

* Add sentencepiece to transformers install for test

* Add matching req to training
  • Loading branch information
rwightman authored Jun 23, 2024
1 parent 45b43c9 commit 2b1e883
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 9 deletions.
4 changes: 2 additions & 2 deletions requirements-test.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
pytest-split==0.8.0
pytest==7.2.0
transformers
timm>=0.9.8
transformers[sentencepiece]
timm>=1.0.3
4 changes: 2 additions & 2 deletions requirements-training.txt
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,6 @@ tqdm
pandas
braceexpand
huggingface_hub
transformers
timm>=0.9.8
transformers[sentencepiece]
timm>=1.0.3
fsspec
2 changes: 0 additions & 2 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,4 @@ regex
ftfy
tqdm
huggingface_hub
sentencepiece
protobuf
timm
7 changes: 4 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,14 +34,15 @@ def _read_reqs(relpath):
# 3 - Alpha
# 4 - Beta
# 5 - Production/Stable
'Development Status :: 3 - Alpha',
'Development Status :: 4 - Beta',
'Intended Audience :: Education',
'Intended Audience :: Science/Research',
'License :: OSI Approved :: Apache Software License',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
'Programming Language :: Python :: 3.10',
'Programming Language :: Python :: 3.11',
'Programming Language :: Python :: 3.12',
'Topic :: Scientific/Engineering',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
'Topic :: Software Development',
Expand All @@ -58,5 +59,5 @@ def _read_reqs(relpath):
extras_require={
"training": TRAINING_REQUIREMENTS,
},
python_requires='>=3.7',
python_requires='>=3.8',
)

0 comments on commit 2b1e883

Please sign in to comment.