Fix broken SAM after huggingface#39120 (huggingface#39289)

fix

Author: yonigozlan

src/transformers/models/sam/modeling_sam.py

@@ -364,7 +364,7 @@ def forward(
         keys = keys + attn_out
 
         keys = self.layer_norm4(keys)
-        return query, keys, attn_out
+        return queries, keys, attn_out
 
 
 class SamTwoWayTransformer(nn.Module):

src/transformers/models/sam_hq/modeling_sam_hq.py

@@ -787,7 +787,7 @@ def forward(
         keys = keys + attn_out
 
         keys = self.layer_norm4(keys)
-        return query, keys, attn_out
+        return queries, keys, attn_out
 
 
 class SamHQTwoWayTransformer(nn.Module):