Skip to content

Commit e6e3088

Browse files
authored
update tensor-rt llm in enum (#390)
* update tensor-rt llm in enum * fix to be the same as in the egp and spellbook-backend
1 parent bf1db14 commit e6e3088

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

clients/python/llmengine/data_types.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ class LLMInferenceFramework(str, Enum):
1616
TEXT_GENERATION_INFERENCE = "text_generation_inference"
1717
VLLM = "vllm"
1818
LIGHTLLM = "lightllm"
19+
TENSORRT_LLM = "tensorrt-llm"
1920

2021

2122
class LLMSource(str, Enum):

0 commit comments

Comments
 (0)