diff --git a/api/onnx_web/convert/diffusion/original.py b/api/onnx_web/convert/diffusion/original.py
index dbcaa330..3a304183 100644
--- a/api/onnx_web/convert/diffusion/original.py
+++ b/api/onnx_web/convert/diffusion/original.py
@@ -1650,6 +1650,7 @@ def extract_checkpoint(
     logger.info(result_status)
 
 
+@torch.no_grad()
 def convert_diffusion_original(
     ctx: ConversionContext,
     model: ModelDict,
diff --git a/api/onnx_web/convert/diffusion/textual_inversion.py b/api/onnx_web/convert/diffusion/textual_inversion.py
index cf8deb68..cef2206e 100644
--- a/api/onnx_web/convert/diffusion/textual_inversion.py
+++ b/api/onnx_web/convert/diffusion/textual_inversion.py
@@ -11,6 +11,7 @@ from ..utils import ConversionContext
 logger = getLogger(__name__)
 
 
+@torch.no_grad()
 def convert_diffusion_textual_inversion(
     context: ConversionContext, name: str, base_model: str, inversion: str
 ):
diff --git a/api/onnx_web/worker/context.py b/api/onnx_web/worker/context.py
index 1b14d80e..f9eb25b3 100644
--- a/api/onnx_web/worker/context.py
+++ b/api/onnx_web/worker/context.py
@@ -74,14 +74,22 @@ class WorkerContext:
 
 
 class JobStatus:
+    name: str
+    device: str
+    progress: int
+    cancelled: bool
+    finished: bool
+
     def __init__(
         self,
         name: str,
+        device: DeviceParams,
         progress: int = 0,
         cancelled: bool = False,
         finished: bool = False,
     ) -> None:
         self.name = name
+        self.device = device.device
         self.progress = progress
         self.cancelled = cancelled
         self.finished = finished