huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-07-27 02:10:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

updated default value

Browse Source
This commit is contained in:
Mohit Sharma 2024-09-27 12:39:12 +00:00
parent ac2dccd174
commit a24c2cc5e9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/text_generation_server/layers/attention/rocm.py
View File

@ -283,7 +283,7 @@ elif ENGINE == "triton":
softmax_scale: float,
window_size_left: int = -1,
causal: bool = True,
softcap: float = 0.0,
softcap: Optional[float] = None,
):
if softcap is not None:
raise NotImplementedError("softcap is only available with CK flash attn")