mirror of
https://github.com/huggingface/trl.git
synced 2025-10-20 18:43:52 +08:00
feat: add requirements (module and repo)
This commit is contained in:
11
requirements.txt
Normal file
11
requirements.txt
Normal file
@ -0,0 +1,11 @@
|
||||
|
||||
-e .
|
||||
jupyterlab==2.0.1
|
||||
nbdev==0.2.16
|
||||
numpy==1.18.2
|
||||
pandas==1.0.3
|
||||
simpletransformers==0.21.4
|
||||
torch==1.4.0
|
||||
tqdm==4.43.0
|
||||
transformers==2.6.0
|
||||
wandb==0.8.31
|
@ -20,7 +20,7 @@ license = apache2
|
||||
status = 2
|
||||
|
||||
# Optional. Same format as setuptools requirements
|
||||
# requirements =
|
||||
requirements = torch>=1.4.0, transformers>=2.6.0, numpy>=1.18.2
|
||||
# Optional. Same format as setuptools console_scripts
|
||||
# console_scripts =
|
||||
|
||||
|
Reference in New Issue
Block a user