Skip to content

Commit

Permalink
nits
Browse files Browse the repository at this point in the history
  • Loading branch information
ronghanghu committed Dec 11, 2024
1 parent c692a77 commit b3eadda
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions sam2/sam2_video_predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -980,9 +980,7 @@ def __init__(self, *args, **kwargs):
self._compile_all_components()

def _compile_all_components(self):
print(
"Compiling all components for for vos setting. First time may be very slow."
)
print("Compiling all components for VOS setting. First time may be very slow.")
self.memory_encoder.forward = torch.compile(
self.memory_encoder.forward,
mode="max-autotune",
Expand Down

0 comments on commit b3eadda

Please sign in to comment.