huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-05-24 04:22:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix the crash in default ATTENTION path for Gaudi backend (#3235)

Browse Source 
Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
This commit is contained in:
Wang, Yi 2025-05-20 20:02:32 +08:00 committed by GitHub
parent 000e313a92
commit 43b1b07fb9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backends/gaudi/server/text_generation_server/tgi_service.py
View File

@ -31,6 +31,7 @@ def main(args):
trust_remote_code=args.trust_remote_code, trust_remote_code=args.trust_remote_code,
uds_path=args.uds_path, uds_path=args.uds_path,
max_input_tokens=args.max_input_tokens, max_input_tokens=args.max_input_tokens,
kv_cache_dtype="auto",
) )