roop/core
Shivam Kumar 9c1ea4fa01
Fix GPU not being used even when it available
On lots of systems onnxruntime doesn't detect GPU unless pytorch is imported before it. So despite having CUDA and CUDNN setup correctly, it is only using CPUExecutionProvider. By importing pytorch first, the issue is fixed. So let's use this until any official solution is available.

See: https://stackoverflow.com/questions/75294639/onnxruntime-inference-with-cudnn-on-gpu-only-working-if-pytorch-imported-first
2023-05-31 09:20:41 +05:30
..
__init__.py initial build 2023-05-28 20:19:40 +05:30
config.py Create face analyser and face swapper instance on demand 2023-05-30 16:29:58 +02:00
globals.py Fix GPU not being used even when it available 2023-05-31 09:20:41 +05:30
processor.py Update processor.py 2023-05-31 02:44:58 +02:00
utils.py Improves final output video quality significantly 2023-05-30 19:07:20 +03:00