michael/scrypted
1
0
Fork 0
mirror of https://github.com/koush/scrypted.git synced 2026-09-16 16:50:40 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

onnx: fix platform check

Browse Source
This commit is contained in:
Koushik Dutta
2024-09-05 10:34:22 -07:00
parent 28f1ce9d4e
commit e23c9c22dc
4 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/onnx/package-lock.json generated
View File

@@ -1,12 +1,12 @@
{
"name": "@scrypted/onnx",
"version": "0.1.109",
"version": "0.1.110",
"lockfileVersion": 2,
"requires": true,
"packages": {
"": {
"name": "@scrypted/onnx",
"version": "0.1.109",
"version": "0.1.110",
"devDependencies": {
"@scrypted/sdk": "file:../../sdk"
}

2
plugins/onnx/package.json
View File

@@ -42,5 +42,5 @@
"devDependencies": {
"@scrypted/sdk": "file:../../sdk"
},
"version": "0.1.109"
"version": "0.1.110"
}

2
plugins/onnx/src/ort/__init__.py
View File

@@ -94,7 +94,7 @@ class ONNXPlugin(
if sys.platform == 'darwin':
providers.append("CoreMLExecutionProvider")
if ('linux' in sys.platform or 'win' in sys.platform) and platform.machine() == 'x86_64':
if ('linux' in sys.platform or 'win' in sys.platform) and (platform.machine() == 'x86_64' or platform.machine() == 'AMD64'):
deviceId = int(deviceId)
providers.append(("CUDAExecutionProvider", { "device_id": deviceId }))

4
server/package-lock.json generated
View File

@@ -1,12 +1,12 @@
{
"name": "@scrypted/server",
"version": "0.117.2",
"version": "0.117.3",
"lockfileVersion": 3,
"requires": true,
"packages": {
"": {
"name": "@scrypted/server",
"version": "0.117.2",
"version": "0.117.3",
"hasInstallScript": true,
"license": "ISC",
"dependencies": {