review comments

Signed-off-by: Alexander Matveev <amatveev@redhat.com>
This commit is contained in:
Alexander Matveev
2025-01-28 23:01:54 +00:00
parent 1ccf100c6a
commit 39c4a4cdb5

View File

@ -556,9 +556,6 @@ class TPUModelRunner(ModelRunnerBase):
assert self.model is not None
self.model(token_ids, position_ids, None, kv_caches)
def profile_run(self) -> None:
raise NotImplementedError()
def capture_model(self) -> None:
"""Compile the model."""