mirror of
https://github.com/meta-llama/llama-stack.git
synced 2026-01-08 00:01:28 +00:00
fix: Add missing shutdown handler for TorchtunePostTrainingImpl
Without it, when server shuts down, it posts the following warning: __main__:129 server: No shutdown method for TorchtunePostTrainingImpl Signed-off-by: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>
This commit is contained in:
parent
bc8daf7fea
commit
f7c1e87293
1 changed files with 3 additions and 0 deletions
|
|
@ -43,6 +43,9 @@ class TorchtunePostTrainingImpl:
|
|||
self.jobs = {}
|
||||
self.checkpoints_dict = {}
|
||||
|
||||
async def shutdown(self):
|
||||
pass
|
||||
|
||||
async def supervised_fine_tune(
|
||||
self,
|
||||
job_uuid: str,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue