Skip to content

Commit f02e4a5

Browse files
committed
chore: remove SAM2 reference
1 parent 753915b commit f02e4a5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/_compile.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -597,7 +597,7 @@ def save(
597597
output_format (str): Format to save the model. Options include exported_program | torchscript.
598598
retrace (bool): When the module type is a fx.GraphModule, this option re-exports the graph using torch.export.export(strict=False) to save it.
599599
This flag is experimental for now.
600-
pickle_protocol (int): The pickle protocol to use to save the model. Default is 2. Increase this to 4 or higher for large models (eg: SAM2)
600+
pickle_protocol (int): The pickle protocol to use to save the model. Default is 2. Increase this to 4 or higher for large models
601601
"""
602602
if isinstance(module, CudaGraphsTorchTensorRTModule):
603603
module = module.compiled_module

0 commit comments

Comments
 (0)