Revert "Update for failing test"

This reverts commit de61c1d.
microsoft · Nov 6, 2024 · 2c5740a · 2c5740a
1 parent b8287fa
commit 2c5740a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepspeed/inference/v2/model_implementations/layer_container_base.py b/deepspeed/inference/v2/model_implementations/layer_container_base.py
@@ -14,6 +14,7 @@
 
 # Currently have dependency loops for the type hints.
 InferenceModel = Type["InferenceModel"]
+LayerContainer = Type["LayerContainer"]
 
 MAPPING_KEY = "PARAM_MAPPING"
 PLIST_HELPERS = "_ds_plist_strip_vals"