storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-09 00:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix minor bug with function call that doesn't exist.

Browse Source
This commit is contained in:
CrossProduct
2025-01-07 23:32:06 +00:00
parent 7b569daf55
commit 5bd5bc4a59
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/utils/memory_utils.py
+1 -1
View File
@@ -21,7 +21,7 @@ def get_memory_statistics(precision: int = 3) -> Dict[str, Any]:
max_memory_allocated = torch.cuda.max_memory_allocated(device)
max_memory_reserved = torch.cuda.max_memory_reserved(device)
elif torch.mps.is_available():
elif torch.backends.mps.is_available():
memory_allocated = torch.mps.current_allocated_memory()
else: