phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 12:07:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore:Added a check for if dpo_output_dir exists in config

Browse source
This commit is contained in:
sarthakdeshpande 2025-08-19 22:46:17 +05:30
parent 5ce2f00650
commit 75bdc7b169
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/inline/post_training/huggingface/post_training.py
View file

@ -135,7 +135,7 @@ class HuggingFacePostTrainingImpl:
resources_allocated, checkpoints = await recipe.train( resources_allocated, checkpoints = await recipe.train(
model=finetuned_model, model=finetuned_model,
output_dir=f"{self.config.dpo_output_dir}/{job_uuid}", output_dir=f"{self.config.dpo_output_dir}/{job_uuid}" if self.config.dpo_output_dir else None,
job_uuid=job_uuid, job_uuid=job_uuid,
dpo_config=algorithm_config, dpo_config=algorithm_config,
config=training_config, config=training_config,