upagge/roop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

max-memory support for Mac

Browse Source
This commit is contained in:
henryruhs 2023-05-30 13:53:59 +02:00
parent 0f3bc95913
commit aa3158576a
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
run.py
View File

@ -46,12 +46,12 @@ if os.name == "nt":
def limit_resources(): def limit_resources():
if args['max_memory'] >= 1: if args['max_memory'] >= 1:
memory = args['max_memory'] * 1024 * 1024 * 1024 memory = args['max_memory'] * 1024 * 1024 * 1024
if str(platform.system()).lower() == 'linux':
import resource
resource.setrlimit(resource.RLIMIT_DATA, (memory, memory))
if str(platform.system()).lower() == 'windows': if str(platform.system()).lower() == 'windows':
import win32api import win32api
win32api.SetProcessWorkingSetSize(-1, memory, memory) win32api.SetProcessWorkingSetSize(-1, memory, memory)
else:
import resource
resource.setrlimit(resource.RLIMIT_DATA, (memory, memory))
def pre_check(): def pre_check():