torch~=2.4.0 | |
torchvision~=0.19.1 | |
lightning~=2.1.2 | |
wandb~=0.17.0 | |
hydra-core~=1.3.2 | |
omegaconf~=2.3.0 | |
torchmetrics[image]==0.11.4 | |
wandb-osh==1.2.1 | |
gluonts[torch]==0.13.1 | |
pytorchvideo~=0.1.5 | |
colorama | |
tqdm | |
opencv-python | |
matplotlib | |
click | |
moviepy==1.0.3 | |
imageio | |
einops | |
pandas | |
pyzmq | |
pyrealsense2 | |
internetarchive | |
h5py | |
rotary_embedding_torch | |
diffusers | |
timm |