mirror of
https://github.com/s0md3v/roop.git
synced 2024-08-24 08:31:17 +08:00
Double GPU performance
This commit is contained in:
parent
b0a7b19b1b
commit
1f0c3b4bb5
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user