allenai / longformer

Longformer: The Long-Document Transformer
https://arxiv.org/abs/2004.05150
Apache License 2.0
2k stars 268 forks source link

"requirements.txt" update (transformers==3.0.2) #258

Open XinyiYing opened 6 months ago

XinyiYing commented 6 months ago

"requirements.txt" should be updated as follows:

transformers==3.0.2
pytorch-lightning
torch>=1.6.0
tensorboardX
test-tube==0.7.5
nlp
rouge_score

Please use the following commands for using longformer:

pip install -r requirements.txt
python setup.py develop
SouLeo commented 3 months ago

Hi @XinyiYing , could you clarify which python and cudatoolkit versions we should initialize our conda environment with for this new requirement file?

Thanks!

lu-ym commented 1 month ago

Is the cuda version 10 series?