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. (#195)

Browse Source 
Co-authored-by: Aryan <aryan@huggingface.co>
This commit is contained in:
ArEnSc
2025-01-07 18:55:27 -05:00
committed by GitHub
parent f1b02964d7
commit 905c22edd8
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: