diff --git a/paddlespeech/s2t/exps/u2/bin/quant.py b/paddlespeech/s2t/exps/u2/bin/quant.py old mode 100644 new mode 100755 index 71101e1c4..f686d003d --- a/paddlespeech/s2t/exps/u2/bin/quant.py +++ b/paddlespeech/s2t/exps/u2/bin/quant.py @@ -159,12 +159,12 @@ class U2Infer(): # jit save logger.info(f"export save: {self.args.export_path}") - self.ptq.save_quantized_model( - self.model, - self.args.export_path, - postprocess=False, - combine_params=True, - skip_forward=True) + self.ptq.ptq._convert(self.model) + paddle.jit.save( + self.model, + self.args.export_path, + combine_params=True, + skip_forward=True) def main(config, args): diff --git a/setup.py b/setup.py index c1757b194..35668bddb 100644 --- a/setup.py +++ b/setup.py @@ -75,7 +75,7 @@ base = [ "braceexpand", "pyyaml", "pybind11", - "paddleslim==2.4.0", + "paddleslim==2.3.4", ] server = ["fastapi", "uvicorn", "pattern_singleton", "websockets"]