diff --git a/modules/torch_utils.py b/modules/torch_utils.py index a07e02853b1..97afef25cec 100644 --- a/modules/torch_utils.py +++ b/modules/torch_utils.py @@ -21,6 +21,6 @@ def get_param(model) -> torch.nn.Parameter: def float64(t: torch.Tensor): """return torch.float64 if device is not mps or xpu, else return torch.float32""" match t.device.type: - case 'mps', 'xpu': + case 'mps' | 'xpu': return torch.float32 return torch.float64