Fix (brevitas_examples/llm): remove unnecessary checkpointing (#1161)

Xilinx · Jan 15, 2025 · 9721379 · 9721379
1 parent a7efcba
commit 9721379
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/src/brevitas_examples/llm/main.py b/src/brevitas_examples/llm/main.py
@@ -517,10 +517,6 @@ def quantize_llm(args):
             apply_bias_correction(model, calibration_loader)
             print("Bias correction applied.")
 
-            if args.checkpoint_name is not None:
-                print(f"Saving checkpoint to {args.checkpoint_name}")
-                torch.save(model.state_dict(), args.checkpoint_name)
-
         if args.eval and not args.no_quantize:
             print("Model eval...")
             with torch.no_grad(), quant_inference_mode(model):