small fix

microsoft · Nov 9, 2023 · 91385d2 · 91385d2
1 parent 27ba203
commit 91385d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/inference/v2/engine_factory.py b/deepspeed/inference/v2/engine_factory.py
@@ -21,7 +21,7 @@
 from .model_implementations.flat_model_helpers import make_metadata_filename, ModelMetadata
 
 def buid_engine_from_ds_checkpoint(path:str, engine_config: RaggedInferenceEngineConfig,
-                    debug_level: int = logging.INFO) -> InferenceEngineV2::
+                    debug_level: int = logging.INFO) -> InferenceEngineV2:
 
     inference_logger(level=debug_level)
     # Load metadata, for grabbing the policy name we'll have all ranks just check for