using paddleslim 2.3.4 and paddle 2.4

pull/2568/head
Hui Zhang 3 years ago
parent 2389ed6675
commit fa027a832e

@ -159,12 +159,12 @@ class U2Infer():
# jit save # jit save
logger.info(f"export save: {self.args.export_path}") logger.info(f"export save: {self.args.export_path}")
self.ptq.save_quantized_model( self.ptq.ptq._convert(self.model)
self.model, paddle.jit.save(
self.args.export_path, self.model,
postprocess=False, self.args.export_path,
combine_params=True, combine_params=True,
skip_forward=True) skip_forward=True)
def main(config, args): def main(config, args):

@ -75,7 +75,7 @@ base = [
"braceexpand", "braceexpand",
"pyyaml", "pyyaml",
"pybind11", "pybind11",
"paddleslim==2.4.0", "paddleslim==2.3.4",
] ]
server = ["fastapi", "uvicorn", "pattern_singleton", "websockets"] server = ["fastapi", "uvicorn", "pattern_singleton", "websockets"]

Loading…
Cancel
Save