Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

updated default value

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, softmax_scale: float,
window_size_left: int = -1, window_size_left: int = -1,
causal: bool = True, causal: bool = True,
softcap: float = 0.0, softcap: Optional[float] = None,
): ):
if softcap is not None: if softcap is not None:
raise NotImplementedError("softcap is only available with CK flash attn") raise NotImplementedError("softcap is only available with CK flash attn")