Commit Graph

36 Commits

Author SHA1 Message Date
Somdev Sangwan
4867f356ef
fix gpu issues for some envs 2023-05-31 12:43:43 +05:30
Somdev Sangwan
1cc34ce8e0 bruh moment 12 2023-05-31 12:42:34 +05:30
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
Henry Ruhs
6258f3c084
Update processor.py 2023-05-31 02:44:58 +02:00
henryruhs
5feb6b0d71 Resolve absolute model path 2023-05-30 21:52:39 +02:00
henryruhs
98a5111d53 Undo: Resolve absolute model path 2023-05-30 21:48:06 +02:00
henryruhs
ffc2bd2485 Resolve absolute model path 2023-05-30 21:12:19 +02:00
Henry Ruhs
cfa9c6cfe9
Merge branch 'main' into main 2023-05-30 20:38:46 +02:00
Furkan Gözükara
561397ab05
Improves final output video quality significantly
Improves final output video quality significantly
2023-05-30 19:07:20 +03:00
henryruhs
80fe3721ee Remove debug output 2023-05-30 16:36:03 +02:00
Henry Ruhs
c10e37cd2f
Merge branch 'main' into main 2023-05-30 16:33:02 +02:00
henryruhs
41c69d318d Create face analyser and face swapper instance on demand 2023-05-30 16:29:58 +02:00
Somdev Sangwan
90e2ba0fc7 properly name output file when its an image 2023-05-30 19:07:22 +05:30
Henry Ruhs
0d2fbca4ce
Merge branch 'main' into main 2023-05-30 15:32:07 +02:00
xavi
42eaab90ac ffmpeg params for good audio sync 2023-05-30 11:55:31 +02:00
henryruhs
365f3d790a Introduce pre_check(), Cleanup requirements.txt 2023-05-30 09:01:03 +02:00
henryruhs
2b14613a46 Move process time to the correct place 2023-05-30 01:28:16 +02:00
henryruhs
dc0653ad39 Restore globals, add process time for better comparison 2023-05-30 01:03:43 +02:00
henryruhs
54f800dd0e Add GPU support, Quit on missing model, Remove globals (sorry) 2023-05-30 00:40:02 +02:00
Somdev Sangwan
4bb6ea416e reverse providers logic 2023-05-29 23:10:52 +05:30
Somdev Sangwan
976354427c
fix windows issues fr 2023-05-29 22:53:45 +05:30
Somdev Sangwan
de34810b85 fix windows paths, finally ig 2023-05-29 21:24:40 +05:30
Somdev Sangwan
f94a79a885 testing in prod 2023-05-29 20:49:11 +05:30
Somdev Sangwan
a25265e031 bruh moment 5, actually 2023-05-29 19:52:51 +05:30
Somdev Sangwan
7041bad118
Merge pull request #28 from henryruhs/main
Skip processing if there is no face, print output to console
2023-05-29 19:31:47 +05:30
henryruhs
e452f755be Add final empty line 2023-05-29 16:00:57 +02:00
henryruhs
31de5fde57 Skip processing if there is no face, print output to console 2023-05-29 15:47:57 +02:00
Somdev Sangwan
32ebd9191c bruh moment 3 2023-05-29 19:13:55 +05:30
Somdev Sangwan
fa813d8025 handle when user has gpu but won't use it 2023-05-29 19:04:22 +05:30
Somdev Sangwan
499bd2e3f6
bruh moment 2 2023-05-29 18:52:16 +05:30
Somdev Sangwan
b6efc0bed8
handle no-face cases 2023-05-29 17:50:26 +05:30
Somdev Sangwan
ae3b2f8d3c
fetch available providers 2023-05-29 17:39:52 +05:30
Somdev Sangwan
c18f66697a
fetch available providers 2023-05-29 17:39:29 +05:30
Somdev Sangwan
bc8fe94eda
bruh moment 2023-05-29 16:40:36 +05:30
Somdev Sangwan
0c867779a1 prompt user for savefile 2023-05-29 13:39:44 +05:30
Somdev Sangwan
f5d85163e2
initial build 2023-05-28 20:19:40 +05:30