fix: rerun linter

2024-10-09 20:10:31 +00:00 · 2024-10-09 20:10:31 +00:00 · 99b1cf5948
parent c396c54231
commit 99b1cf5948
1 changed files with 2 additions and 0 deletions
--- a/server/tests/models/test_flash_causal_lm.py
+++ b/server/tests/models/test_flash_causal_lm.py
@ -23,9 +23,11 @@ set_adapter_to_index({})


 if SYSTEM == "cuda":
+
    def test_flash_causal_lm_warmup():
        flash_causal_lm_warmup()

+
 def flash_causal_lm_warmup():
    revision = "main"
    quantize = None