Initial commit

2026-04-13 18:03:33 -07:00
commit 73d16c9b69
26 changed files with 1660 additions and 0 deletions
@@ -0,0 +1,14 @@
+
+import torch
+import platform
+
+def detect_device():
+    if torch.cuda.is_available():
+        return "cuda"
+    if platform.system() == "Darwin":
+        try:
+            if torch.backends.mps.is_available():
+                return "mps"
+        except:
+            pass
+    return "cpu"