mirror of https://github.com/bmaltais/kohya_ss
Merge pull request #3266 from jim60105/fix/typo
fix(validate_requirements): correct typopull/3274/head
commit
08f2ddcf82
|
|
@ -135,9 +135,9 @@ def log_cuda_info(torch):
|
||||||
|
|
||||||
def log_mps_info(torch):
|
def log_mps_info(torch):
|
||||||
"""Log information about Apple Silicone (MPS)"""
|
"""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(
|
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.')
|
log.warning('MPS support is still experimental, proceed with caution.')
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue