From 4142d3fd2c5fa78147980befdffbca2a0b9ec34d Mon Sep 17 00:00:00 2001 From: Laikh Tewari Date: Tue, 7 May 2024 00:01:34 -0700 Subject: [PATCH] typo fix in doc on saving models (#2818) --- docsrc/user_guide/saving_models.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docsrc/user_guide/saving_models.rst b/docsrc/user_guide/saving_models.rst index 73fee6e23c..42cff7b954 100644 --- a/docsrc/user_guide/saving_models.rst +++ b/docsrc/user_guide/saving_models.rst @@ -35,7 +35,7 @@ Here's an example usage inputs = [torch.randn((1, 3, 224, 224)).cuda()] # trt_ep is a torch.fx.GraphModule object trt_gm = torch_tensorrt.compile(model, ir="dynamo", inputs) - torchtrt.save(trt_gm, "trt.ep", inputs=inputs) + torch_tensorrt.save(trt_gm, "trt.ep", inputs=inputs) # Later, you can load it and run inference model = torch.export.load("trt.ep").module() @@ -97,4 +97,4 @@ Here's an example usage # file_path can be trt.ep or trt.ts file obtained via saving the model (refer to the above section) inputs = [torch.randn((1, 3, 224, 224)).cuda()] model = torch_tensorrt.load().module() - model(*inputs) \ No newline at end of file + model(*inputs)