zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 09:21:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix incorrect type annotation (#720)

Browse Source 
A `Tuple` is missing in this type annotation.
This commit is contained in:
Kevin Wang 2024-04-24 18:52:43 -04:00 committed by GitHub
parent abcd891851
commit 8a265f0d54
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llms/speculative_decoding/model.py
View File

@ -105,7 +105,7 @@ class MultiHeadAttention(nn.Module):
values: mx.array, values: mx.array,
mask: Optional[mx.array], mask: Optional[mx.array],
cache: Optional[Tuple[mx.array, mx.array]] = None, cache: Optional[Tuple[mx.array, mx.array]] = None,
) -> [mx.array, Tuple[mx.array, mx.array]]: ) -> Tuple[mx.array, Tuple[mx.array, mx.array]]:
queries = self.query_proj(queries) queries = self.query_proj(queries)
keys = self.key_proj(keys) keys = self.key_proj(keys)
values = self.value_proj(values) values = self.value_proj(values)