Merge pull request #16088 from cuba3/dev_cuba3

Maintaining Project Compatibility for Python 3.9 Users Without Upgrade Requirements.
2024-07-06 09:52:40 +03:00 · 2024-07-06 09:52:40 +03:00 · edebe4d4de
parent b9c3f4ec2c 9e60cdbc3f
commit edebe4d4de
1 changed files with 2 additions and 3 deletions
--- a/modules/torch_utils.py
+++ b/modules/torch_utils.py
@ -20,7 +20,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':
-            return torch.float32
+    if t.device.type in ['mps', 'xpu']:
+        return torch.float32
    return torch.float64