423A35C7/sam2
1
0
Fork 0
mirror of https://github.com/facebookresearch/sam2.git synced 2025-11-04 11:32:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

nits

Browse Source
This commit is contained in:
Ronghang Hu 2024-12-11 19:27:27 +00:00
parent c692a7743d
commit b3eaddab6f
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
sam2/sam2_video_predictor.py
View File

@ -980,9 +980,7 @@ class SAM2VideoPredictorVOS(SAM2VideoPredictor):
self._compile_all_components() self._compile_all_components()
def _compile_all_components(self): def _compile_all_components(self):
print( print("Compiling all components for VOS setting. First time may be very slow.")
"Compiling all components for for vos setting. First time may be very slow."
)
self.memory_encoder.forward = torch.compile( self.memory_encoder.forward = torch.compile(
self.memory_encoder.forward, self.memory_encoder.forward,
mode="max-autotune", mode="max-autotune",