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

fix: rerun linter

Browse Source
This commit is contained in:
David Holtz 2024-10-09 20:10:31 +00:00
parent c396c54231
commit 99b1cf5948
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/tests/models/test_flash_causal_lm.py
View File

@ -23,9 +23,11 @@ set_adapter_to_index({})
if SYSTEM == "cuda": if SYSTEM == "cuda":
def test_flash_causal_lm_warmup(): def test_flash_causal_lm_warmup():
flash_causal_lm_warmup() flash_causal_lm_warmup()
def flash_causal_lm_warmup(): def flash_causal_lm_warmup():
revision = "main" revision = "main"
quantize = None quantize = None