Hotfix torch installation (#83)

* Hotfix torch installation

* Hotfix torch installation

.github/workflows/ci.yml

@@ -30,5 +30,6 @@ jobs:
     uses: actions/setup-python@v2
     with:
      python-version: '3.10'
-  - run: pip install -r requirements-ci.txt
+  - run: pip install -r requirements.txt
+  - run: pip install pytest
   - run: pytest

facefusion/installer.py

@@ -35,6 +35,10 @@ def run() -> None:
 		onnxruntime_key = answers['onnxruntime_key']
 		onnxruntime_name, onnxruntime_version = ONNXRUNTIMES[onnxruntime_key]
 		python_id = 'cp' + str(sys.version_info.major) + str(sys.version_info.minor)
+		subprocess.call([ 'pip', 'uninstall', 'torch', '-y' ])
+		if onnxruntime_key == 'cuda':
+			subprocess.call([ 'pip', 'install', '-r', 'requirements.txt', '--extra-index-url', 'https://download.pytorch.org/whl/cu118' ])
+		else:
 			subprocess.call([ 'pip', 'install', '-r', 'requirements.txt' ])
 		if onnxruntime_key != 'cpu':
 			subprocess.call([ 'pip', 'uninstall', 'onnxruntime', onnxruntime_name, '-y' ])

requirements-ci.txt

@@ -1,11 +0,0 @@
-insightface==0.7.3
-numpy==1.24.3
-onnx==1.14.1
-onnxruntime==1.15.1
-opencv-python==4.8.0.76
-opennsfw2==0.10.2
-protobuf==4.24.2
-pytest==7.4.0
-psutil==5.9.5
-tensorflow==2.13.0
-tqdm==4.66.1

requirements.txt

@@ -1,5 +1,3 @@
---extra-index-url https://download.pytorch.org/whl/cu118
-
 gfpgan==1.3.8
 gradio==3.42.0
 insightface==0.7.3