fix floating point mem limit

OpenAgentsInc · Apr 26, 2024 · ce79304 · ce79304
1 parent 4d1db6b
commit ce79304
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -18,7 +18,7 @@ class Runner (JobRunner):
 
     def __init__(self, filters, meta, template, sockets):
         super().__init__(filters, meta, template, sockets)
-        self.MAX_MEMORY_CACHE_GB = int(os.getenv('MAX_MEMORY_CACHE_GB', self.MAX_MEMORY_CACHE_GB))
+        self.MAX_MEMORY_CACHE_GB = float(os.getenv('MAX_MEMORY_CACHE_GB', self.MAX_MEMORY_CACHE_GB))
 
 
     async def deserializeFromBlob(self,  url,  out_vectors , out_content):