[CI][RFC] Replace black-jupyter with ruff-format (#37210)
Co-authored-by: Alison Shao <a.shao@wustl.edu>
This commit is contained in:
co-authored by
Alison Shao
parent
2641e427be
commit
28262c20df
@@ -74,7 +74,9 @@ class _FakeMHATokenToKVPool(_FakeKVCache):
|
||||
self.v_head_dim = (
|
||||
swa_v_head_dim
|
||||
if swa_v_head_dim is not None
|
||||
else v_head_dim if v_head_dim is not None else head_dim
|
||||
else v_head_dim
|
||||
if v_head_dim is not None
|
||||
else head_dim
|
||||
)
|
||||
self._create_buffers()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user