Increase maximum macOS version to 15.0.1 when forcing upcast attention (#5191)

This commit is contained in:
Jonathan Avila 2024-10-09 20:21:41 -06:00 committed by GitHub
parent 25eac1d780
commit 4b2f0d9413
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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