From 019bd3792ca1aa12b5df34bebeba99b128b0998e Mon Sep 17 00:00:00 2001 From: Sergii Dymchenko Date: Mon, 26 Aug 2024 13:22:47 -0700 Subject: [PATCH] Fix wrong docstring arg name --- xformers/ops/fmha/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/xformers/ops/fmha/__init__.py b/xformers/ops/fmha/__init__.py index 61ce63b021..f40e0910b1 100644 --- a/xformers/ops/fmha/__init__.py +++ b/xformers/ops/fmha/__init__.py @@ -677,7 +677,7 @@ def merge_attentions( either as a list of tensors of shapes [B, G, H, M] or [B, H, M] or as a single tensor of shape [num_chunks, B, G, H, M] or [num_chunks, B, H, M] write_lse: whether to output LSE - out_dype: dtype of attn_out + output_dtype: dtype of attn_out Returns: attn_out: [B, M, G, H, Kq] or [B, M, H, Kq]