fix pokeman

Signed-off-by: changwangss <[email protected]>
intel · Jun 21, 2024 · b2e0a8d · b2e0a8d
1 parent 3aac3a4
commit b2e0a8d
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/examples/huggingface/pytorch/text-to-image/quantization/ptq/run_diffusion.py b/examples/huggingface/pytorch/text-to-image/quantization/ptq/run_diffusion.py
@@ -319,8 +319,6 @@ def eval_func(model):
                     accuracy_criterion=accuracy_criterion
                 )
                 os.makedirs(args.output_dir, exist_ok=True)
-                model.config.torch_dtype = "int8"
-                model.config.save_pretrained(args.output_dir)
                 model = fit(model,
                             quantization_config,
                            eval_func=eval_func,