diff --git a/openllm-python/src/openllm/_llm.pyi b/openllm-python/src/openllm/_llm.pyi
index d38632e6..d4c62bd6 100644
--- a/openllm-python/src/openllm/_llm.pyi
+++ b/openllm-python/src/openllm/_llm.pyi
@@ -73,7 +73,7 @@ class LLM(Generic[M, T]):
     serialisation: LiteralSerialisation = ...,
     trust_remote_code: bool = ...,
     embedded: bool = ...,
-    torch_dtype: Dtype = ...,
+    dtype: Dtype = ...,
     low_cpu_mem_usage: bool = ...,
     **attrs: Any,
   ) -> None: ...