mirror of https://github.com/bmaltais/kohya_ss
fix(validate_requirements): correct typo
- Fixes a typo in the variable name and log message from "reccomended" to the correct "recommended" when logging Apple MPS memory information. Signed-off-by: CHEN, CHUN <jim60105@gmail.com>pull/3266/head
parent
40556e9397
commit
ae0f3dbf46
|
|
@ -135,9 +135,9 @@ def log_cuda_info(torch):
|
|||
|
||||
def log_mps_info(torch):
|
||||
"""Log information about Apple Silicone (MPS)"""
|
||||
max_reccomended_mem = round(torch.mps.recommended_max_memory() / 1024**2)
|
||||
max_recommended_mem = round(torch.mps.recommended_max_memory() / 1024**2)
|
||||
log.info(
|
||||
f"Torch detected Apple MPS: {max_reccomended_mem}MB Unified Memory Available"
|
||||
f"Torch detected Apple MPS: {max_recommended_mem}MB Unified Memory Available"
|
||||
)
|
||||
log.warning('MPS support is still experimental, proceed with caution.')
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue