fix memory leak

2023-06-28 07:20:31 -05:00 · 2023-06-28 07:20:31 -05:00 · f06444c3cb
parent 5fc3e4c2e8
commit f06444c3cb
1 changed files with 1 additions and 1 deletions
--- a/scripts/spartan/Worker.py
+++ b/scripts/spartan/Worker.py
@ -281,7 +281,7 @@ class Worker:
                    free_vram = int(memory_response['free']) / (1024 * 1024 * 1024)
                    total_vram = int(memory_response['total']) / (1024 * 1024 * 1024)
                    logger.debug(f"Worker '{self.uuid}' {free_vram:.2f}/{total_vram:.2f} GB VRAM free\n")
-                    self.free_vram = bytes(memory_response['free'])
+                    self.free_vram = memory_response['free']
                except KeyError:
                    error = memory_response['cuda']['error']
                    logger.debug(f"CUDA doesn't seem to be available for worker '{self.uuid}'\nError: {error}")