Initial commit

requirements.txt

@@ -1,21 +1,27 @@
-torch>=2.0.0
-torchvision>=0.15.0
-numpy>=1.24.0
-pillow>=9.5.0
-diffusers>=0.27.0
-transformers==4.57.0
-accelerate>=0.20.0
-opencv-python>=4.8.0
-rembg>=2.0.50
-facenet-pytorch>=2.5.3
-einops>=0.7.0
+torch==2.4.0
+torchvision==0.19.0
+numpy==1.26.4
+pillow==10.4.0
+diffusers[torch]==0.30.3
+transformers==4.44.2
+accelerate==0.33.0
+opencv-python==4.10.0.84
+rembg
+facenet-pytorch
+einops==0.8.0
 ninja
-plyfile>=0.9
-easydict>=1.10
-pyyaml>=6.0
-lpips>=0.1.4
-huggingface-hub>=0.19.0
+plyfile==1.0.3
+easydict==1.13
+pyyaml==6.0.2
+lpips==0.1.4
+huggingface-hub==0.35.3
 videoio==0.3.0
-gradio>=5.0.0
-matplotlib>=3.7.0
+gradio==5.49.1
+matplotlib==3.7.5
 onnxruntime>=1.15.0
+xformers==0.0.27.post2
+scikit-image==0.21.0
+jaxtyping==0.2.19
+pytorch-msssim==1.0.0
+ffmpeg-python==0.2.0
+tqdm