Fix usage example

pytorch · Dec 22, 2024 · bf5fa71 · bf5fa71
1 parent 25075e2
commit bf5fa71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docsrc/user_guide/saving_models.rst b/docsrc/user_guide/saving_models.rst
@@ -34,7 +34,7 @@ Here's an example usage
     model = MyModel().eval().cuda()
     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) 
+    trt_gm = torch_tensorrt.compile(model, ir="dynamo", inputs=inputs) 
     torch_tensorrt.save(trt_gm, "trt.ep", inputs=inputs)
 
     # Later, you can load it and run inference