Increase maximum macOS version to 15.0.1 when forcing upcast attention

This commit is contained in:
Jonathan Avila 2024-10-09 00:57:40 -06:00
parent 203942c8b2
commit e8034b62f8

View File

@ -899,7 +899,7 @@ def force_upcast_attention_dtype():
upcast = args.force_upcast_attention
try:
macos_version = tuple(int(n) for n in platform.mac_ver()[0].split("."))
if (14, 5) <= macos_version < (14, 7): # black image bug on recent versions of MacOS
if (14, 5) <= macos_version <= (15, 0, 1): # black image bug on recent versions of macOS
upcast = True
except:
pass