We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2f1df87 commit e12d88dCopy full SHA for e12d88d
py/torch_tensorrt/_compile.py
@@ -534,4 +534,4 @@ def save(
534
exp_program = torch.export.export(
535
module, tuple(arg_inputs), kwargs=kwarg_inputs, strict=False
536
)
537
- torch.export.save(exp_program, file_path)
+ torch.export.save(exp_program, file_path)
0 commit comments