requirements

requirements.txt

@@ -1,26 +1,52 @@
-decord==0.6.0
-einops==0.3.0
-numpy==1.24.2
-omegaconf==2.1.1
-pandas==2.0.0
-Pillow==9.5.0
-pytorch_lightning==1.9.0
-PyYAML==6.0
-setuptools==65.6.3
+Pytorch-Lightning==1.9.0
+decord
+kornia
+timm
+open_clip_torch
+av
+omegaconf
+transformers
+einops
+scikit-learn
+taming-transformers-rom1504
+pandas
+triton
 xformers==0.0.16
 torch==1.13.1
 torchvision
-tqdm==4.65.0
-transformers==4.25.1
-moviepy
-av
-timm
-scikit-learn 
-open_clip_torch==2.12.0
-kornia
-gradio==3.37.0
+fairscale
+gradio==3.37.0 #3.35.2
 plotly
 imageio==2.14.1
 imageio-ffmpeg==0.4.7
 opencv-python==4.8.0.74
-uvicorn==0.24.0.post1
+moviepy
+Pillow
+tqdm
+
+# decord==0.6.0
+# einops==0.3.0
+# numpy==1.24.2
+# omegaconf==2.1.1
+# pandas==2.0.0
+# Pillow==9.5.0
+# pytorch_lightning==1.9.0
+# PyYAML==6.0
+# setuptools==65.6.3
+# xformers==0.0.16
+# torch==1.13.1
+# torchvision
+# tqdm==4.65.0
+# transformers==4.25.1
+# moviepy
+# av
+# timm
+# scikit-learn 
+# open_clip_torch==2.12.0
+# kornia
+# gradio==3.37.0
+# plotly
+# imageio==2.14.1
+# imageio-ffmpeg==0.4.7
+# opencv-python==4.8.0.74
+# uvicorn==0.24.0.post1