remove duplicated qkv computation in na_vit_nested_tensor_3d.py (#341)

Author: JacobLinCool

vit_pytorch/na_vit_nested_tensor_3d.py

@@ -83,17 +83,6 @@ def forward(
 
         # split heads
 
-        def split_heads(t):
-            return t.unflatten(-1, (self.heads, self.dim_head)).transpose(1, 2).contiguous()
-
-        # queries, keys, values
-
-        query = self.to_queries(x)
-        key = self.to_keys(context)
-        value = self.to_values(context)
-
-        # split heads
-
         def split_heads(t):
             return t.unflatten(-1, (self.heads, self.dim_head))