stable-diffusion-webui/modules/devices.py

import torch

# has_mps is only available in nightly pytorch (for now), `getattr` for compatibility
has_mps = getattr(torch, 'has_mps', False)

cpu = torch.device("cpu")


def get_optimal_device():
    if torch.cuda.is_available():
        return torch.device("cuda")

    if has_mps:
        return torch.device("mps")

    return cpu
Modular device management 2022-09-11 13:11:27 +08:00			`import torch`

			# has_mps is only available in nightly pytorch (for now), `getattr` for compatibility
			`has_mps = getattr(torch, 'has_mps', False)`

CLIP interrogator 2022-09-11 23:48:36 +08:00			`cpu = torch.device("cpu")`


Modular device management 2022-09-11 13:11:27 +08:00			`def get_optimal_device():`
CLIP interrogator 2022-09-11 23:48:36 +08:00			`if torch.cuda.is_available():`
			`return torch.device("cuda")`

			`if has_mps:`
			`return torch.device("mps")`

			`return cpu`