From bb4ed8b53cf3984c551395c4785c3f44d30f7960 Mon Sep 17 00:00:00 2001
From: Aaron <29749331+aarnphm@users.noreply.github.com>
Date: Sat, 9 Dec 2023 09:59:02 -0500
Subject: [PATCH] fix(llm): correct annotations definitions

Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
---
 openllm-python/src/openllm/_llm.pyi | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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: ...