From 9bfb75aa472f97a0c8a7713f88fd4725d3c2bc13 Mon Sep 17 00:00:00 2001 From: shengdinghu Date: Thu, 12 May 2022 06:31:53 +0800 Subject: [PATCH] update requirements --- examples/setup_seq2seq.py | 13 +++---------- requirements.txt | 2 +- 2 files changed, 4 insertions(+), 11 deletions(-) diff --git a/examples/setup_seq2seq.py b/examples/setup_seq2seq.py index 0aa5a64..acecd77 100755 --- a/examples/setup_seq2seq.py +++ b/examples/setup_seq2seq.py @@ -1,5 +1,5 @@ """Install Compacter.""" -import os +import os import setuptools from torch.utils.cpp_extension import BuildExtension, CUDAExtension @@ -29,17 +29,10 @@ def setup_package(): 'Programming Language :: Python :: 3.7.10', ], 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}, install_requires=[ + "pyarrow==7.0.0", + "datasets==1.17.0" ], ) diff --git a/requirements.txt b/requirements.txt index a1e2702..59d4d55 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ torch>=1.8.0 transformers>=4.10.0 -datasets>=1.17.0 +datasets==1.17.0 sentencepiece>=0.1.96 tqdm>=4.62.2 loralib