Skip to content

Commit 6605b8b

Browse files
committed
remove _set_model_layer_prefix()
Signed-off-by: shanjiaz <[email protected]>
1 parent 3cac826 commit 6605b8b

File tree

1 file changed

+0
-14
lines changed

1 file changed

+0
-14
lines changed

src/llmcompressor/core/lifecycle.py

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,6 @@ def initialize(
9898
logger.debug("Initializing compression lifecycle")
9999
self.recipe_container.append(recipe, recipe_stage, recipe_args)
100100
self.modifiers = self.recipe_container.get_modifiers()
101-
self._set_model_layer_prefix()
102101

103102
mod_data = []
104103
for mod in self.modifiers:
@@ -229,16 +228,3 @@ def _validate_event_order(self, event_type: EventType) -> bool:
229228
if valid:
230229
self._last_event_type = event_type
231230
return valid
232-
233-
def _set_model_layer_prefix(self):
234-
compiled_recipe = self.recipe_container.compiled_recipe
235-
if (
236-
compiled_recipe is None
237-
or (metadata := compiled_recipe.metadata) is None
238-
or (model_metadata := metadata.target_model) is None
239-
):
240-
return False
241-
242-
self.state.model.layer_prefix = model_metadata.layer_prefix
243-
logger.debug("Model layer prefix set to {}", self.state.model.layer_prefix)
244-
return True

0 commit comments

Comments
 (0)