fix(llm): correct annotations definitions

Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
Aaron
2023-12-09 09:59:02 -05:00
parent c0307c9d71
commit bb4ed8b53c

View File

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