# PyTorch: https://pytorch.org/get-started/locally/
# For newest CUDA, just do the following:
# pip3 install torch torchvision torchaudio

# HuggingFace
transformers
accelerate
datasets
evaluate
sacrebleu
rouge_score
peft
Pillow

seaborn
scikit-learn
tqdm
loguru
deepspeed
wandb

# For mistral, we need transformers=4.36.2 as in https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2/discussions/19
# PyTorch: https://pytorch.org/get-started/locally/
