Use 'torch.backends.mps.is_available' to check if mps is supported (#3164)

2023-07-18 08:27:18 +08:00 · 2023-07-18 08:27:18 +08:00 · 89e0d15cf5
commit 89e0d15cf5
parent 234c58ccd1
3 changed files with 4 additions and 4 deletions
--- a/modules/text_generation.py
+++ b/modules/text_generation.py
@ -57,7 +57,7 @@ def encode(prompt, add_special_tokens=True, add_bos_token=True, truncation_lengt
        return input_ids.numpy()
    elif shared.args.deepspeed:
        return input_ids.to(device=local_rank)
-    elif torch.has_mps:
+    elif torch.backends.mps.is_available():
        device = torch.device('mps')
        return input_ids.to(device)
    else: