diff --git a/python/tvm/driver/tvmc/frontends.py b/python/tvm/driver/tvmc/frontends.py index 21d3d59fb013..13ab3dd170c3 100644 --- a/python/tvm/driver/tvmc/frontends.py +++ b/python/tvm/driver/tvmc/frontends.py @@ -262,7 +262,9 @@ def load(self, path, shape_dict=None, **kwargs): input_shapes = list(shape_dict.items()) logger.debug("parse Torch model and convert into Relay computation graph") - return relay.frontend.from_pytorch(traced_model, input_shapes, **kwargs) + return relay.frontend.from_pytorch( + traced_model, input_shapes, keep_quantized_weight=True, **kwargs + ) class PaddleFrontend(Frontend):