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

Differential Revision: D90237984

fbshipit-source-id: 526fd760f303bf31be4f743bdcd77760496de0de
This commit is contained in:
generatedunixname89002005307016
2026-01-07 05:16:41 -08:00
committed by meta-codesync[bot]
parent 5eb25fb54b
commit 7b89b8fc3f
155 changed files with 310 additions and 0 deletions

View File

@@ -1,4 +1,6 @@
# Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
# pyre-unsafe
from .mask_decoder import MaskDecoder
from .prompt_encoder import PromptEncoder
from .transformer import TwoWayTransformer

View File

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

View File

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

View File

@@ -1,5 +1,7 @@
# Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
# pyre-unsafe
from typing import Any, Optional, Tuple, Type
import numpy as np

View File

@@ -1,5 +1,7 @@
# Copyright (c) Meta Platforms, Inc. and affiliates. All Rights Reserved
# pyre-unsafe
"""
Adapted from:
1. https://github.com/meta-llama/codellama/blob/main/llama/model.py

View File

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