Duplicate import

microsoft · Nov 13, 2023 · d187c3f · d187c3f
1 parent e69abe0
commit d187c3f
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/deepspeed/inference/v2/engine_factory.py b/deepspeed/inference/v2/engine_factory.py
@@ -7,7 +7,6 @@
 import logging
 import os
 import pickle
-from typing import Any
 from packaging import version
 
 from .engine_v2 import InferenceEngineV2
@@ -96,7 +95,6 @@ def build_hf_engine(path: str,
         elif model_config.model_type == "llama":
             policy = Llama2Policy(model_config, checkpoint_engine=checkpoint_engine)
         elif model_config.model_type == "mistral":
-            from .model_implementations.mistral.policy import MistralPolicy
             # Ensure we're using the correct version of transformers for mistral
             import transformers
             assert version.parse(transformers.__version__) >= version.parse("4.34.0"), \