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
import os
# reducing num threads doubles performance of gpu-mode
os.environ['OMP_NUM_THREADS'] = '1'
import platform
import signal
import sys
import shutil
import glob
import argparse
import os
import torch
from pathlib import Path
import tkinter as tk
@ -16,7 +18,6 @@ from tkinter.filedialog import asksaveasfilename
import webbrowser
import cv2
import threading
from threadpoolctl import threadpool_limits
from PIL import Image, ImageTk
import roop.globals