fix: fix export in torchscript format.

GdoongMathew · lanluo-nvidia · commit aa2f917a7e5a · 2025-09-30T08:20:53.000-07:00
diff --git a/py/torch_tensorrt/_compile.py b/py/torch_tensorrt/_compile.py
@@ -660,7 +660,7 @@ def save(
             "Input model is of type nn.Module. Saving nn.Module directly is not supported. Supported model types torch.jit.ScriptModule | torch.fx.GraphModule | torch.export.ExportedProgram."
         )
     elif module_type == _ModuleType.ts:
-        if not all([output_format == f for f in ["exported_program", "aot_inductor"]]):
+        if output_format != "torchscript":
             raise ValueError(
                 "Provided model is a torch.jit.ScriptModule but the output_format specified is not torchscript. Other output formats are not supported"
             )

Original file line number	Diff line number	Diff line change
`@@ -660,7 +660,7 @@ def save(`
`660`	`660`	`"Input model is of type nn.Module. Saving nn.Module directly is not supported. Supported model types torch.jit.ScriptModule \| torch.fx.GraphModule \| torch.export.ExportedProgram."`
`661`	`661`	`)`
`662`	`662`	`elif module_type == _ModuleType.ts:`
`663`		`- if not all([output_format == f for f in ["exported_program", "aot_inductor"]]):`
	`663`	`+ if output_format != "torchscript":`
`664`	`664`	`raise ValueError(`
`665`	`665`	`"Provided model is a torch.jit.ScriptModule but the output_format specified is not torchscript. Other output formats are not supported"`
`666`	`666`	`)`