init

vllm-project · Feb 19, 2025 · 80a86c7 · 80a86c7
1 parent 4a34d0c
commit 80a86c7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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()
+