MLR-Copilot / requirements.txt
Lim0011's picture
Update requirements.txt
1f3a846 verified
raw
history blame contribute delete
505 Bytes
--extra-index-url https://download.pytorch.org/whl/cu118
torch
torchvision
torchaudio
huggingface_hub>=0.12.0
# images
pillow
# common ml packages
transformers>=4.30.0
scikit-learn
xgboost
keras
scipy
numpy==1.26.4
pandas
matplotlib
seaborn
tqdm
kaggle
# llm
anthropic
sentencepiece
peft>=0.3.0
accelerate
evaluate
# graph
torch-geometric==2.3.1
ogb
#others
dacite
python-dotenv
# for p2m specifically
datasets
gradio==3.38
bert_score
tevatron
mdtex2html
retriv
tiktoken
charset-normalizer==3.1.0