ark-instruct-line-item / requirements.txt
tosi-n7's picture
Upload folder using huggingface_hub
d8ffdc4
appdirs
bert_score
black[jupyter]
flask
gradio
huggingface-hub
IPython
ipyparallel
jsonlines
loralib
pycuda
spacy_fastlang
# triton
numpy>=1.26.0
rouge_score>=0.1.2
fire>=0.5.0
# openai
transformers>=4.34.0
torch>=2.0.0
sentencepiece>=0.1.99
tokenizers>=0.14.0
wandb
accelerate>=0.23.0
datasets>=2.14.5
deepspeed>=0.10.3
peft>=0.5.0
# partial
# gradio
einops>=0.7.0
bitsandbytes>=0.41.1
scipy>=1.11.3
protobuf>=4.24.4