Skip to content

Commit f06444c

Browse files
committed
fix memory leak
1 parent 5fc3e4c commit f06444c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/spartan/Worker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,7 @@ def request(self, payload: dict, option_payload: dict, sync_options: bool):
281281
free_vram = int(memory_response['free']) / (1024 * 1024 * 1024)
282282
total_vram = int(memory_response['total']) / (1024 * 1024 * 1024)
283283
logger.debug(f"Worker '{self.uuid}' {free_vram:.2f}/{total_vram:.2f} GB VRAM free\n")
284-
self.free_vram = bytes(memory_response['free'])
284+
self.free_vram = memory_response['free']
285285
except KeyError:
286286
error = memory_response['cuda']['error']
287287
logger.debug(f"CUDA doesn't seem to be available for worker '{self.uuid}'\nError: {error}")

0 commit comments

Comments
 (0)