remove cuda

meta.json

@@ -1,18 +1,22 @@
 {
-    "module_id": "viam:torch-cpu",
-    "visibility": "public",
-    "url": "https://github.com/viam-labs/torch",
-    "description": "Viam ML Module service serving PyTorch models.",
-    "models": [
-      {
-        "api": "rdk:service:mlmodel",
-        "model": "viam:mlmodel:torch-cpu"
-      }
-    ],
-    "build": {
-        "build": "./build.sh", 
-        "path": "dist/archive.tar.gz", 
-        "arch": ["linux/arm64", "linux/amd64"] 
-      },
-    "entrypoint": "dist/main"
+  "module_id": "viam:torch-cpu",
+  "visibility": "public",
+  "url": "https://github.com/viam-labs/torch",
+  "description": "Viam ML Module service serving PyTorch models.",
+  "models": [
+    {
+      "api": "rdk:service:mlmodel",
+      "model": "viam:mlmodel:torch-cpu"
+    }
+  ],
+  "build": {
+    "build": "./build.sh",
+    "path": "dist/archive.tar.gz",
+    "arch": [
+      "linux/arm64",
+      "linux/amd64",
+      "darwin/arm64"
+    ]
+  },
+  "entrypoint": "dist/main"
 }

requirements.txt

@@ -5,5 +5,12 @@ pylint
 pyinstaller
 google-api-python-client
 pytest
-torchvision
-torch==2.2.1
+torch; sys_platform == 'darwin'
+torch==2.2.2+cpu; sys_platform  == 'linux'
+-f https://download.pytorch.org/whl/cpu/torch_stable.html
+
+torchvision == 0.18; sys_platform  == 'darwin'
+torchvision==0.17.2+cpu;sys_platform  == 'linux'
+-f https://download.pytorch.org/whl/cpu/torch_stable.html
+
+pyinstaller