From 75bdc7b169d0af137066a219880b6452bb3604f1 Mon Sep 17 00:00:00 2001 From: sarthakdeshpande Date: Tue, 19 Aug 2025 22:46:17 +0530 Subject: [PATCH] chore:Added a check for if dpo_output_dir exists in config --- .../providers/inline/post_training/huggingface/post_training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/inline/post_training/huggingface/post_training.py b/llama_stack/providers/inline/post_training/huggingface/post_training.py index 22ace1ae0..e0bf63207 100644 --- a/llama_stack/providers/inline/post_training/huggingface/post_training.py +++ b/llama_stack/providers/inline/post_training/huggingface/post_training.py @@ -135,7 +135,7 @@ class HuggingFacePostTrainingImpl: resources_allocated, checkpoints = await recipe.train( 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, dpo_config=algorithm_config, config=training_config,