From 80a86c7359f6055e59b53a69d7da5431b43ed742 Mon Sep 17 00:00:00 2001 From: George Ohashi Date: Wed, 19 Feb 2025 09:39:39 -0500 Subject: [PATCH] init --- src/llmcompressor/transformers/finetune/runner.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/llmcompressor/transformers/finetune/runner.py b/src/llmcompressor/transformers/finetune/runner.py index 769b84248..d3a7515f6 100644 --- a/src/llmcompressor/transformers/finetune/runner.py +++ b/src/llmcompressor/transformers/finetune/runner.py @@ -288,3 +288,4 @@ def run_sequential_stages(self, checkpoint: Optional[str] = None): torch.cuda.empty_cache() self.trainer.accelerator.free_memory() self.trainer.accelerator.wait_for_everyone() +