using paddleslim 2.3.4 and paddle 2.4

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

@ -159,10 +159,10 @@ 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)
paddle.jit.save(
self.model, self.model,
self.args.export_path, self.args.export_path,
postprocess=False,
combine_params=True, combine_params=True,
skip_forward=True) skip_forward=True)

@ -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