1580158
/
automatic
mirror of https://github.com/vladmandic/automatic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix get_runpod_limit

pull/3819/head
Disty0 2025-03-14 19:44:00 +03:00
parent 4fce7c279e
commit c82d5a8a8a
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/memstats.py
View File

@ -24,6 +24,8 @@ def get_docker_limit():
docker_limit = float(f.read())
except Exception:
docker_limit = sys.float_info.max
if docker_limit == 0:
docker_limit = sys.float_info.max
return docker_limit
@ -31,8 +33,8 @@ def get_runpod_limit():
global runpod_limit # pylint: disable=global-statement
if runpod_limit is not None:
return runpod_limit
runpod_limit = float(os.environ.get('RUNPOD_MEM_GB', sys.float_info.max))
if runpod_limit == 0: # RUNPOD_MEM_GB can be set to 0 for no limit
runpod_limit = float(os.environ.get('RUNPOD_MEM_GB', 0)) * 1024 * 1024 * 1024
if runpod_limit == 0:
runpod_limit = sys.float_info.max
return runpod_limit