Skip to content

Commit 4be64a8

Browse files
authored
Fix code example in README.md (#3253)
1 parent 1820713 commit 4be64a8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ import torch_tensorrt
6262
model = MyModel().eval().cuda() # define your model here
6363
inputs = [torch.randn((1, 3, 224, 224)).cuda()] # define a list of representative inputs here
6464

65-
trt_gm = torch_tensorrt.compile(model, ir="dynamo", inputs)
65+
trt_gm = torch_tensorrt.compile(model, ir="dynamo", inputs=inputs)
6666
torch_tensorrt.save(trt_gm, "trt.ep", inputs=inputs) # PyTorch only supports Python runtime for an ExportedProgram. For C++ deployment, use a TorchScript file
6767
torch_tensorrt.save(trt_gm, "trt.ts", output_format="torchscript", inputs=inputs)
6868
```

0 commit comments

Comments
 (0)