p04798526
/
OpenDeltaMirror
mirror of https://github.com/thunlp/OpenDelta.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update requirements

This commit is contained in:
shengdinghu 2022-05-12 06:31:53 +08:00
parent 211dce1b9d
commit 9bfb75aa47
2 changed files with 4 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
examples/setup_seq2seq.py
View File

@ -1,5 +1,5 @@
"""Install Compacter.""" """Install Compacter."""
import os import os
import setuptools import setuptools
from torch.utils.cpp_extension import BuildExtension, CUDAExtension from torch.utils.cpp_extension import BuildExtension, CUDAExtension
@ -29,17 +29,10 @@ def setup_package():
'Programming Language :: Python :: 3.7.10', 'Programming Language :: Python :: 3.7.10',
], ],
keywords='text nlp machinelearning', keywords='text nlp machinelearning',
# ext_modules=[
# CUDAExtension('seq2seq.projections.fwh_cuda',
# sources=[
# 'seq2seq/projections/fwh_cuda/fwh_cpp.cpp',
# 'seq2seq/projections/fwh_cuda/fwh_cu.cu',
# ]
# )
# ]
# ,
cmdclass={"build_ext": BuildExtension}, cmdclass={"build_ext": BuildExtension},
install_requires=[ install_requires=[
"pyarrow==7.0.0",
"datasets==1.17.0"
], ],
) )

2
requirements.txt
View File

@ -1,6 +1,6 @@
torch>=1.8.0 torch>=1.8.0
transformers>=4.10.0 transformers>=4.10.0
datasets>=1.17.0 datasets==1.17.0
sentencepiece>=0.1.96 sentencepiece>=0.1.96
tqdm>=4.62.2 tqdm>=4.62.2
loralib loralib