apps/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2024-08-24 08:31:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Double GPU performance

Browse Source
This commit is contained in:
henryruhs 2023-06-03 13:37:15 +02:00
parent b0a7b19b1b
commit 1f0c3b4bb5
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
roop/core.py
View File

@ -1,12 +1,14 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import os
# reducing num threads doubles performance of gpu-mode
os.environ['OMP_NUM_THREADS'] = '1'
import platform import platform
import signal import signal
import sys import sys
import shutil import shutil
import glob import glob
import argparse import argparse
import os
import torch import torch
from pathlib import Path from pathlib import Path
import tkinter as tk import tkinter as tk
@ -16,7 +18,6 @@ from tkinter.filedialog import asksaveasfilename
import webbrowser import webbrowser
import cv2 import cv2
import threading import threading
from threadpoolctl import threadpool_limits
from PIL import Image, ImageTk from PIL import Image, ImageTk
import roop.globals import roop.globals