From 7a956978ec701a5ff26885e9460b099178e071f8 Mon Sep 17 00:00:00 2001 From: changwangss Date: Fri, 12 Jul 2024 01:03:42 -0700 Subject: [PATCH] remove pdb Signed-off-by: changwangss --- .../code-generation/quantization/run_generation_cpu_woq.py | 1 - 1 file changed, 1 deletion(-) diff --git a/examples/huggingface/pytorch/code-generation/quantization/run_generation_cpu_woq.py b/examples/huggingface/pytorch/code-generation/quantization/run_generation_cpu_woq.py index a5c282acd33..6934d8b55b3 100644 --- a/examples/huggingface/pytorch/code-generation/quantization/run_generation_cpu_woq.py +++ b/examples/huggingface/pytorch/code-generation/quantization/run_generation_cpu_woq.py @@ -388,7 +388,6 @@ # save model if args.output_dir is not None and ((args.woq or args.load_in_4bit or args.load_in_8bit) and not args.use_neural_speed): - import pdb;pdb.set_trace(); user_model.save_pretrained(args.output_dir) tokenizer.save_pretrained(args.output_dir) # to validate woq model accuracy