Skip to content

Commit

Permalink
remove attribute
Browse files Browse the repository at this point in the history
Signed-off-by: Kyle Sayers <[email protected]>
  • Loading branch information
kylesayrs committed Feb 14, 2025
1 parent ffbec46 commit 8276ece
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 4 deletions.
3 changes: 0 additions & 3 deletions src/llmcompressor/core/lifecycle.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,6 @@ class CompressionLifecycle:
modifiers: List[StageModifiers] = field(default_factory=list)
event_lifecycle: Optional[EventLifecycle] = None

initialized_structure: bool = False
initialized_: bool = False
finalized: bool = False
event_called: bool = False
Expand All @@ -69,7 +68,6 @@ def reset(self):
self.modifiers = []
self.event_lifecycle = None

self.initialized_structure = False
self.initialized_ = False
self.finalized = False
self.event_called = False
Expand All @@ -96,7 +94,6 @@ def pre_initialize_structure(self, **kwargs) -> List[Any]:
if data is not None:
mod_data.append(data)

self.initialized_structure = True
applied_stage_names = [mod.unique_id for mod in self.modifiers if mod.applied]
self.recipe_container.update_applied_stages(applied_stage_names)
logger.info(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -64,5 +64,4 @@ def test_mixin_session_init(mixin_trainer):
mixin_trainer.initialize_session(epoch=0.0, checkpoint=None)
session = active_session()

assert not session.lifecycle.initialized_structure
assert session.lifecycle.initialized_

0 comments on commit 8276ece

Please sign in to comment.