--index-url https://download.pytorch.org/whl/cu121
--extra-index-url https://pypi.org/simple

torch==2.5.1+cu121
torchvision==0.20.1+cu121

transformers==4.44.2
accelerate==1.10.1
peft==0.17.0
safetensors==0.4.5
huggingface_hub[cli,hf-transfer]==0.35.3
datasets==4.1.1
sentencepiece==0.2.0
einops==0.8.0
bitsandbytes==0.43.3

numpy==2.0.2
pandas==2.2.3
pyarrow==21.0.0
tqdm==4.66.5

opencv-python-headless==4.10.0.84
pillow==10.4.0
av==15.1.0
imageio==2.36.0
imageio-ffmpeg==0.5.1

hydra-core==1.3.2
omegaconf==2.3.0
pyyaml==6.0.2
packaging==24.2
python-dotenv==1.0.1
wandb==0.21.1