phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-01 16:24:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

XXX

Browse source 
Signed-off-by: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>
This commit is contained in:
Ihar Hrachyshka 2025-05-20 13:06:35 -04:00
parent 9607140e24
commit 01fdae0adb
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/integration/post_training/test_post_training.py
View file

@ -107,7 +107,8 @@ class TestPostTraining:
logger.info(f"Starting training job with UUID: {self.job_uuid}") logger.info(f"Starting training job with UUID: {self.job_uuid}")
# train with HF trl SFTTrainer as the default # train with HF trl SFTTrainer as the default
os.makedirs("~/.llama/checkpoints/", exist_ok=True) checkpoint_dir = os.path.expanduser("/mnt/")
# os.makedirs(checkpoint_dir, exist_ok=True)
started = datetime.now(timezone.utc) started = datetime.now(timezone.utc)
_ = llama_stack_client.post_training.supervised_fine_tune( _ = llama_stack_client.post_training.supervised_fine_tune(
@ -117,7 +118,7 @@ class TestPostTraining:
training_config=training_config, training_config=training_config,
hyperparam_search_config={}, hyperparam_search_config={},
logger_config={}, logger_config={},
checkpoint_dir="~/.llama/checkpoints/", checkpoint_dir=checkpoint_dir,
) )
while True: while True: