From caf040439adea0884c1df81e7ae0db9c68d92bf5 Mon Sep 17 00:00:00 2001 From: Sergii Dymchenko Date: Thu, 22 Aug 2024 15:14:49 -0700 Subject: [PATCH] Remove non-existing argument from docstring --- segment_anything_fast/modeling/image_encoder.py | 1 - 1 file changed, 1 deletion(-) diff --git a/segment_anything_fast/modeling/image_encoder.py b/segment_anything_fast/modeling/image_encoder.py index 540cd78..369c6de 100644 --- a/segment_anything_fast/modeling/image_encoder.py +++ b/segment_anything_fast/modeling/image_encoder.py @@ -346,7 +346,6 @@ def add_decomposed_rel_pos( Calculate decomposed Relative Positional Embeddings from :paper:`mvitv2`. https://github.com/facebookresearch/mvit/blob/19786631e330df9f3622e5402b4a419a263a2c80/mvit/models/attention.py # noqa B950 Args: - attn (Tensor): attention map. q (Tensor): query q in the attention layer with shape (B, q_h * q_w, C). rel_pos_h (Tensor): relative position embeddings (Lh, C) for height axis. rel_pos_w (Tensor): relative position embeddings (Lw, C) for width axis.