Remove (direct) torch and torchvision dependencies

s0md3v · Jul 30, 2023 · 4c9cf80 · 4c9cf80
1 parent eb2440c
commit 4c9cf80
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 10 deletions.
diff --git a/requirements-ci.txt b/requirements-ci.txt
@@ -6,8 +6,6 @@ psutil==5.9.5
 tk==0.1.0
 customtkinter==5.2.0
 tkinterdnd2==0.3.0
-torch==2.0.1
-torchvision==0.15.2
 onnxruntime==1.15.0
 tensorflow==2.13.0
 opennsfw2==0.10.2

diff --git a/requirements.txt b/requirements.txt
@@ -10,10 +10,6 @@ customtkinter==5.2.0
 tkinterdnd2==0.3.0; sys_platform != 'darwin' and platform_machine != 'arm64'
 tkinterdnd2-universal==1.7.3; sys_platform == 'darwin' and platform_machine == 'arm64'
 pillow==10.0.0
-torch==2.0.1+cu118; sys_platform != 'darwin'
-torch==2.0.1; sys_platform == 'darwin'
-torchvision==0.15.2+cu118; sys_platform != 'darwin'
-torchvision==0.15.2; sys_platform == 'darwin'
 onnxruntime==1.15.1; python_version != '3.9' and sys_platform == 'darwin' and platform_machine != 'arm64'
 onnxruntime-coreml==1.13.1; python_version == '3.9' and sys_platform == 'darwin' and platform_machine != 'arm64'
 onnxruntime-silicon==1.13.1; sys_platform == 'darwin' and platform_machine == 'arm64'

diff --git a/roop/core.py b/roop/core.py
@@ -13,12 +13,8 @@
 import signal
 import shutil
 import argparse
-import torch
 import onnxruntime
-if not 'CUDAExecutionProvider' in onnxruntime.get_available_providers():
-    del torch
 import tensorflow
-
 import roop.globals
 import roop.metadata
 import roop.ui as ui