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
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.args.export_path,
postprocess=False,
combine_params=True,
skip_forward=True)

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

Loading…
Cancel
Save