Add missing Pyre mode headers] [batch:11/N] [shard:17/N]

Differential Revision: D90237984

fbshipit-source-id: 526fd760f303bf31be4f743bdcd77760496de0de

sam3/model/__init__.py

@@ -1 +1,3 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
+
+# pyre-unsafe

sam3/model/act_ckpt_utils.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import inspect
 from functools import wraps
 from typing import Callable, TypeVar, Union

sam3/model/box_ops.py

@@ -1,4 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
+
+# pyre-unsafe
 """
 Utilities for bounding box manipulation and GIoU.
 """

sam3/model/data_misc.py

@@ -1,4 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
+
+# pyre-unsafe
 """
 Misc functions, including distributed helpers.
 """

sam3/model/decoder.py

@@ -1,4 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
+
+# pyre-unsafe
 """
 Transformer decoder.
 Inspired from Pytorch's version, adds the pre-norm variant

sam3/model/edt.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 """Triton kernel for euclidean distance transform (EDT)"""
 
 import torch

sam3/model/encoder.py

@@ -1,6 +1,8 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 # Based on https://github.com/IDEA-Research/GroundingDINO
 
+# pyre-unsafe
+
 from typing import Any, Dict, List, Optional, Tuple
 
 import torch

sam3/model/geometry_encoders.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 from typing import Tuple
 
 import torch

sam3/model/io_utils.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import contextlib
 import os
 import queue

sam3/model/maskformer_segmentation.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import math
 from typing import Dict, List, Optional
 

sam3/model/memory.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import math
 from typing import Tuple
 

sam3/model/model_misc.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 """Various utility models"""
 
 import copy

sam3/model/necks.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 """Necks are the interface between a vision backbone and the rest of the detection model"""
 
 from copy import deepcopy

sam3/model/position_encoding.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import math
 from typing import Optional
 

sam3/model/sam1_task_predictor.py

@@ -1,6 +1,8 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 # All rights reserved.
 
+# pyre-unsafe
+
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.
 

sam3/model/sam3_image.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import os
 from copy import deepcopy
 from typing import Dict, List, Optional, Tuple

sam3/model/sam3_image_processor.py

@@ -1,4 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
+
+# pyre-unsafe
 from typing import Dict, List
 
 import numpy as np

sam3/model/sam3_tracker_base.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import logging
 
 import torch

sam3/model/sam3_tracker_utils.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import numpy as np
 import torch
 import torch.nn.functional as F

sam3/model/sam3_tracking_predictor.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import logging
 from collections import OrderedDict
 

sam3/model/sam3_video_base.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import datetime
 import logging
 import math

sam3/model/sam3_video_inference.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import logging
 from collections import defaultdict
 

sam3/model/sam3_video_predictor.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 import datetime
 import gc
 import multiprocessing as mp

sam3/model/text_encoder_ve.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 from collections import OrderedDict
 from typing import Callable, List, Optional, Tuple, Union
 

sam3/model/tokenizer_ve.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 """
 Text Tokenizer.
 

sam3/model/utils/__init__.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 # All rights reserved.
 
+# pyre-unsafe
+
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.

sam3/model/utils/misc.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 from collections import defaultdict
 from dataclasses import fields, is_dataclass
 from typing import Any, Mapping, Protocol, runtime_checkable

sam3/model/utils/sam1_utils.py

@@ -1,6 +1,8 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 # All rights reserved.
 
+# pyre-unsafe
+
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.
 

sam3/model/utils/sam2_utils.py

@@ -1,6 +1,8 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 # All rights reserved.
 
+# pyre-unsafe
+
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.
 

sam3/model/vitdet.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 """
 ViTDet backbone adapted from Detectron2.
 This module implements Vision Transformer (ViT) backbone for object detection.

sam3/model/vl_combiner.py

@@ -1,5 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
 
+# pyre-unsafe
+
 """Provides utility to combine a vision backbone with a language backbone."""
 
 from copy import copy