comments

Author: horheynm

tests/llmcompressor/transformers/obcq/test_consecutive_runs.py

@@ -41,7 +41,6 @@ def _test_consecutive_runs(
                 recipe=self.first_recipe,
                 output_dir=self.output_first,
                 oneshot_device=self.device,
-                clear_sparse_session=False,
             )
 
             first_model = AutoModelForCausalLM.from_pretrained(
@@ -60,7 +59,6 @@ def _test_consecutive_runs(
             session_recipe = session.lifecycle.recipe_container.compiled_recipe
             stages = [stage.group for stage in session_recipe.stages]
             self.assertEqual(len(stages), 1)
-            # session.reset()
 
         with create_session():
             recipe = infer_recipe_from_model_path(model_path=self.output_first)
@@ -75,7 +73,6 @@ def _test_consecutive_runs(
                 recipe=self.second_recipe,
                 output_dir=self.output_second,
                 oneshot_device=self.device,
-                clear_sparse_session=False,
             )
 
             second_model = AutoModelForCausalLM.from_pretrained(