diff --git a/invokeai/backend/util/devices.py b/invokeai/backend/util/devices.py index 84ca7ee02b..0a65f6c67a 100644 --- a/invokeai/backend/util/devices.py +++ b/invokeai/backend/util/devices.py @@ -1,11 +1,9 @@ from __future__ import annotations -import platform from contextlib import nullcontext from typing import Union import torch -from packaging import version from torch import autocast from invokeai.app.services.config import InvokeAIAppConfig @@ -37,7 +35,7 @@ def choose_precision(device: torch.device) -> str: device_name = torch.cuda.get_device_name(device) if not ("GeForce GTX 1660" in device_name or "GeForce GTX 1650" in device_name): return "float16" - elif device.type == "mps" and version.parse(platform.mac_ver()[0]) < version.parse("14.0.0"): + elif device.type == "mps": return "float16" return "float32"