From 8eab92a7df8a4d835196861fe5638f9eaa2cd5be Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 26 Mar 2023 11:49:58 -0500 Subject: [PATCH] define device on pending job --- api/onnx_web/worker/command.py | 1 + 1 file changed, 1 insertion(+) diff --git a/api/onnx_web/worker/command.py b/api/onnx_web/worker/command.py index 089be538..1b0a1a0a 100644 --- a/api/onnx_web/worker/command.py +++ b/api/onnx_web/worker/command.py @@ -41,6 +41,7 @@ class JobCommand: args: Any, kwargs: dict[str, Any], ): + self.device = device self.name = name self.fn = fn self.args = args