torch
accelerate>=0.29.3
pytest
tensorboard
pynvml
transformers
datasets
peft
torchvision
opencv-python
rich
imageio
