diff --git a/llama_stack/apis/post_training/post_training.py b/llama_stack/apis/post_training/post_training.py index 5385a39b0..8841dc1d0 100644 --- a/llama_stack/apis/post_training/post_training.py +++ b/llama_stack/apis/post_training/post_training.py @@ -30,7 +30,7 @@ class OptimizerType(Enum): @json_schema_type class DatasetFormat(Enum): instruct = "instruct" - chat = "dialog" + dialog = "dialog" @json_schema_type