openvino: fix cluster label to require x64

koush · Nov 23, 2024 · 38c00f5 · 38c00f5
1 parent ab47389
commit 38c00f5
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 2 deletions.
diff --git a/plugins/coreml/.vscode/settings.json b/plugins/coreml/.vscode/settings.json
@@ -1,6 +1,6 @@
 
 {
-    "scrypted.debugHost": "127.0.0.1",
+    "scrypted.debugHost": "scrypted-nvr",
     "python.analysis.extraPaths": [
         "./node_modules/@scrypted/sdk/types/scrypted_python"
     ]

diff --git a/plugins/openvino/src/ov/__init__.py b/plugins/openvino/src/ov/__init__.py
@@ -286,6 +286,9 @@ async def getSettings(self) -> list[Setting]:
             },
         ]
 
+    def getClusterLabels(self):
+        return {"labels": {"require": ["compute", "x64"]}}
+
     async def putSetting(self, key: str, value: SettingValue):
         self.storage.setItem(key, value)
         await self.onDeviceEvent(scrypted_sdk.ScryptedInterface.Settings.value, None)

diff --git a/plugins/openvino/src/ov/face_recognition.py b/plugins/openvino/src/ov/face_recognition.py
@@ -16,6 +16,9 @@
 
 
 class OpenVINOFaceRecognition(FaceRecognizeDetection):
+    def getClusterLabels(self):
+        return {"labels": {"require": ["compute", "x64"]}}
+
     def downloadModel(self, model: str):
         scrypted_yolov9 = "scrypted_yolov9" in model
         ovmodel = "best-converted" if scrypted_yolov9 else "best"

diff --git a/plugins/openvino/src/ov/text_recognition.py b/plugins/openvino/src/ov/text_recognition.py
@@ -15,6 +15,9 @@
 
 
 class OpenVINOTextRecognition(TextRecognition):
+    def getClusterLabels(self):
+        return {"labels": {"require": ["compute", "x64"]}}
+
     def downloadModel(self, model: str):
         ovmodel = "best"
         precision = self.plugin.precision

diff --git a/server/.vscode/launch.json b/server/.vscode/launch.json
@@ -99,7 +99,7 @@
                 "SCRYPTED_CLUSTER_WORKER_NAME": "Macaroni 2",
                 "SCRYPTED_CLUSTER_LABELS": "compute",
                 "SCRYPTED_CLUSTER_MODE": "client",
-                "SCRYPTED_CLUSTER_SERVER": "192.168.2.124",
+                "SCRYPTED_CLUSTER_SERVER": "192.168.2.130",
                 "SCRYPTED_CLUSTER_SECRET": "swordfish",
                 "SCRYPTED_CAN_RESTART": "true",
                 "SCRYPTED_VOLUME": "/Users/koush/.scrypted-cluster/volume-client",