root commited on
Commit
609d3fa
·
1 Parent(s): 329d53d

dependenceis

Browse files
Files changed (1) hide show
  1. requirements.txt +11 -0
requirements.txt CHANGED
@@ -6,6 +6,7 @@ onnx==1.14.0
6
  insightface==0.7.3
7
  psutil==5.9.5
8
  tk==0.1.0
 
9
  torch==2.0.1+cu118; sys_platform != 'darwin'
10
  torch==2.0.1; sys_platform == 'darwin'
11
  torchvision==0.15.2+cu118; sys_platform != 'darwin'
@@ -22,12 +23,15 @@ gfpgan==1.3.8
22
  gradio==3.41.2
23
  onnxruntime-coreml==1.13.1; python_version == '3.9' and sys_platform == 'darwin' and platform_machine != 'arm64'
24
  transformers==4.41.1
 
25
 
26
  # Add additional dependencies
27
  diffusers==0.24.0
28
  omegaconf==2.2.3
29
 
30
  # Face swap related dependencies
 
 
31
 
32
  # Stuff huggingface is complaining about
33
  einops==0.4.1
@@ -37,8 +41,15 @@ av==11.0.0
37
  accelerate==0.21.0
38
  clip @ https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip#sha256=b5842c25da441d6c581b53a5c60e0c2127ebafe0f746f8e15561a006c6c3be6a
39
  decord==0.6.0
 
 
 
 
 
40
  scipy==1.11.4
41
  torchdiffeq==0.2.3
 
 
42
 
43
 
44
  # Additional dependencies for RIFE
 
6
  insightface==0.7.3
7
  psutil==5.9.5
8
  tk==0.1.0
9
+ pillow==9.5.0
10
  torch==2.0.1+cu118; sys_platform != 'darwin'
11
  torch==2.0.1; sys_platform == 'darwin'
12
  torchvision==0.15.2+cu118; sys_platform != 'darwin'
 
23
  gradio==3.41.2
24
  onnxruntime-coreml==1.13.1; python_version == '3.9' and sys_platform == 'darwin' and platform_machine != 'arm64'
25
  transformers==4.41.1
26
+ controlnet-aux==0.0.7
27
 
28
  # Add additional dependencies
29
  diffusers==0.24.0
30
  omegaconf==2.2.3
31
 
32
  # Face swap related dependencies
33
+ facenet-pytorch==2.5.2
34
+ dlib==19.22.0
35
 
36
  # Stuff huggingface is complaining about
37
  einops==0.4.1
 
41
  accelerate==0.21.0
42
  clip @ https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip#sha256=b5842c25da441d6c581b53a5c60e0c2127ebafe0f746f8e15561a006c6c3be6a
43
  decord==0.6.0
44
+ gradio_client==0.5.0
45
+ imageio==2.33.0
46
+ imageio-ffmpeg==0.4.9
47
+ scikit-image==0.21.0
48
+ scikit-learn==1.3.2
49
  scipy==1.11.4
50
  torchdiffeq==0.2.3
51
+ torchmetrics==1.2.1
52
+ torchsde==0.2.5
53
 
54
 
55
  # Additional dependencies for RIFE