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

Merge branch 'main' into feature/full-finetuning

Browse Source
This commit is contained in:
ArEnSc
2025-01-07 19:14:25 -05:00
committed by GitHub
parent cb9381b40e 905c22edd8
commit bd02e6d9d2
3 changed files with 7 additions and 7 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: