From c82d5a8a8a29e5ff631bec81fbbc4ebce48285ea Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 14 Mar 2025 19:44:00 +0300 Subject: [PATCH] Fix get_runpod_limit --- modules/memstats.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/memstats.py b/modules/memstats.py index c4f7f7f49..160492069 100644 --- a/modules/memstats.py +++ b/modules/memstats.py @@ -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