Change the requirements files

s0md3v · Aug 14, 2023 · fb9372c · fb9372c
1 parent 7c27007
commit fb9372c
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 1 deletion.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -30,7 +30,7 @@ jobs:
     uses: actions/setup-python@v2
     with:
      python-version: '3.10'
-  - run: pip install -r requirements-headless.txt
+  - run: pip install -r requirements-ci.txt
   - run: python run.py -s .github/examples/source.jpg -t .github/examples/target.mp4 -o .github/examples/output.mp4
     if: matrix.os != 'windows-latest'
   - run: python run.py -s .github\examples\source.jpg -t .github\examples\target.mp4 -o .github\examples\output.mp4

diff --git a/requirements-headless.txt → requirements-ci.txt b/requirements-headless.txt → requirements-ci.txt
diff --git a/requirements-docker.cpu.txt b/requirements-docker.cpu.txt
@@ -0,0 +1,15 @@
+basicsr==1.4.2
+gfpgan==1.3.8
+gradio==3.39.0
+insightface==0.7.3
+numpy==1.24.3
+onnx==1.14.0
+onnxruntime==1.15.1
+opencv-python==4.8.0.74
+opennsfw2==0.10.2
+pillow==10.0.0
+protobuf==4.23.4
+psutil==5.9.5
+realesrgan==0.3.0
+tensorflow==2.13.0
+tqdm==4.65.0
diff --git a/requirements-docker.cuda.txt b/requirements-docker.cuda.txt
@@ -0,0 +1,17 @@
+--extra-index-url https://download.pytorch.org/whl/cu118
+
+basicsr==1.4.2
+gfpgan==1.3.8
+gradio==3.39.0
+insightface==0.7.3
+numpy==1.24.3
+onnx==1.14.0
+onnxruntime-gpu==1.15.1
+opencv-python==4.8.0.74
+opennsfw2==0.10.2
+pillow==10.0.0
+protobuf==4.23.4
+psutil==5.9.5
+realesrgan==0.3.0
+tensorflow==2.13.0
+tqdm==4.65.0