format

3 years ago · bbdbfed0fb
parent fa027a832e
commit bbdbfed0fb
1 changed files with 4 additions and 4 deletions
--- a/paddlespeech/s2t/exps/u2/bin/quant.py
+++ b/paddlespeech/s2t/exps/u2/bin/quant.py
@ -161,10 +161,10 @@ class U2Infer():
        logger.info(f"export save: {self.args.export_path}")
        self.ptq.ptq._convert(self.model)
        paddle.jit.save(
-                self.model,
+            self.model,
-                self.args.export_path,
+            self.args.export_path,
-                combine_params=True,
+            combine_params=True,
-                skip_forward=True)
+            skip_forward=True)
 def main(config, args):