removed deadcode

octoml · Jan 26, 2024 · cb65a3c · cb65a3c
1 parent 8f51dee
commit cb65a3c
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/serve/mlc_serve/engine/model_module.py b/serve/mlc_serve/engine/model_module.py
@@ -7,7 +7,6 @@
 from .base import ChatMessage, RequestId, MLCServeEngineConfig, RequestState, SequenceId
 from ..model.base import ModelArtifactConfig
 from .sampling_params import SamplingParams
-from .constrained_sampling import JSONLogitsProcessor
 
 
 @dataclass

diff --git a/serve/mlc_serve/model/tvm_model.py b/serve/mlc_serve/model/tvm_model.py
@@ -248,7 +248,6 @@ def generate(
         )
 
         input_shape = input_ids.shape
-        cs_input_ids = input_ids
 
         if self.disco_session:
             input_ids = copy_to_worker_0(self.disco_session, input_ids)