huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-10 11:54:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: fix typo in mpt_modeling.py

Browse Source 
implemetation -> implementation
This commit is contained in:
Ikko Eltociear Ashimine 2023-07-31 11:52:53 +09:00 committed by GitHub
parent 3ef5ffbc64
commit a1cec743ee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/text_generation_server/models/custom_modeling/mpt_modeling.py
View File

@ -297,7 +297,7 @@ def triton_flash_attn_fn(
class MultiheadAttention(nn.Module): class MultiheadAttention(nn.Module):
"""Multi-head self attention. """Multi-head self attention.
Using torch or triton attention implemetation enables user to also use Using torch or triton attention implementation enables user to also use
additive bias. additive bias.
""" """
@ -386,7 +386,7 @@ class MultiheadAttention(nn.Module):
class MultiQueryAttention(nn.Module): class MultiQueryAttention(nn.Module):
"""Multi-Query self attention. """Multi-Query self attention.
Using torch or triton attention implemetation enables user to also use Using torch or triton attention implementation enables user to also use
additive bias. additive bias.
""" """