Spaces:
Runtime error
Runtime error
| # Copyright (c) Meta Platforms, Inc. and affiliates. | |
| # All rights reserved. | |
| # This source code is licensed under the license found in the | |
| # LICENSE file in the root directory of this source tree. | |
| from .efficient_sam import build_efficient_sam | |
| def build_efficient_sam_vitt(checkpoint=None): | |
| return build_efficient_sam( | |
| encoder_patch_embed_dim=192, | |
| encoder_num_heads=3, | |
| checkpoint=checkpoint, | |
| ).eval() | |
| def build_efficient_sam_vits(checkpoint=None): | |
| return build_efficient_sam( | |
| encoder_patch_embed_dim=384, | |
| encoder_num_heads=6, | |
| checkpoint=checkpoint, | |
| ).eval() | |