torch
wandb
numpy
opencv-python
pathlib
typing
torch
tqdm
safetensors
huggingface_hub
datasets
einops
torchvision
streamlit
pillow
matplotlib
transformers