clean

octoml · Nov 16, 2023 · 9bf9c19 · 9bf9c19
1 parent f26d409
commit 9bf9c19
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/serve/mlc_serve/model/paged_cache_model.py b/serve/mlc_serve/model/paged_cache_model.py
@@ -14,11 +14,10 @@
 from tvm.runtime import disco as di
 
 from mlc_llm import utils
-from mlc_llm.relax_model.llama import LlamaConfig
 
 from .base import get_model_artifact_config
 from .tokenizer import HfTokenizerModule
-from ..engine import ChatMessage, RequestId, SamplingType, MLCServeEngineConfig, SamplingParams
+from ..engine import RequestId, SamplingType, MLCServeEngineConfig, SamplingParams
 from ..engine.model_module import (
     DecodeRequest,
     PrefillRequest,