fix

huggingface · Oct 5, 2023 · b3890b8 · b3890b8
1 parent 69eceb5
commit b3890b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/autotrain/cli/run_llm.py b/src/autotrain/cli/run_llm.py
@@ -408,7 +408,7 @@ def __init__(self, args):
                 print(f"Bot: {tgi.chat(prompt)}")
 
         cuda_available = torch.cuda.is_available()
-        mps_available = torch.mps.is_available()
+        mps_available = torch.backends.mps.is_available()
 
         if not cuda_available and not mps_available:
             raise ValueError("No GPU/MPS device found. LLM training requires an accelerator")