Update requirements.txt

requirements.txt

@@ -1,13 +1,38 @@
 torch
 torchvision
+
 pytorch-lightning==1.5.0
-einops
-opencv-python
+omegaconf==2.1.1
+test-tube>=0.7.5
+einops==0.3.0
+kornia==0.6
+timm==0.6.12
+torchmetrics==0.6.0
+
+transformers==4.29.2
+open_clip_torch==2.0.2
+safetensors==0.2.7
+datasets==2.14.5
+
+opencv-python==4.6.0.66
+opencv-contrib-python==4.6.0.66
+opencv-python-headless==4.6.0.66
+
+scipy==1.9.1
+scikit-image==0.20.0
+Pillow
+numpy
+
+matplotlib
+imageio==2.9.0
+imageio-ffmpeg==0.4.2
+
+invisible-watermark>=0.1.5
+webdataset==0.2.5
+prettytable==3.6.0
+addict==2.4.0
+yapf==0.32.0
+
 rembg
-gradio>=4.29.0
-transformers
 huggingface_hub
-numpy
-Pillow
-scipy
-omegaconf==2.1.1
+gradio>=4.29.0