diff --git a/llama_stack/providers/inline/post_training/common/validator.py b/llama_stack/providers/inline/post_training/common/validator.py
index eac6b3302..e4456ea98 100644
--- a/llama_stack/providers/inline/post_training/common/validator.py
+++ b/llama_stack/providers/inline/post_training/common/validator.py
@@ -9,7 +9,7 @@
 #
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
-from llama_stack.apis.common.type_system import StringType
+from llama_stack.apis.common.type_system import ChatCompletionInputType, StringType
 from llama_stack.apis.datasets import Datasets
 from llama_stack.providers.utils.common.data_schema_validator import (
     ColumnName,
@@ -19,7 +19,7 @@ from llama_stack.providers.utils.common.data_schema_validator import (
 EXPECTED_DATASET_SCHEMA = {
     "instruct": [
         {
-            ColumnName.chat_completion_input.value: StringType(),
+            ColumnName.chat_completion_input.value: ChatCompletionInputType(),
             ColumnName.expected_answer.value: StringType(),
         }
     ],
diff --git a/llama_stack/templates/experimental-post-training/build.yaml b/llama_stack/templates/experimental-post-training/build.yaml
index aa7695bca..912248628 100644
--- a/llama_stack/templates/experimental-post-training/build.yaml
+++ b/llama_stack/templates/experimental-post-training/build.yaml
@@ -13,6 +13,7 @@ distribution_spec:
     post_training:
     - inline::torchtune
     datasetio:
+    - inline::localfs
     - remote::huggingface
     telemetry:
     - inline::meta-reference
diff --git a/llama_stack/templates/experimental-post-training/run.yaml b/llama_stack/templates/experimental-post-training/run.yaml
index 308f03a2e..b89f86816 100644
--- a/llama_stack/templates/experimental-post-training/run.yaml
+++ b/llama_stack/templates/experimental-post-training/run.yaml
@@ -29,6 +29,9 @@ providers:
     provider_type: inline::basic
     config: {}
   datasetio:
+  - provider_id: huggingface-0
+    provider_type: remote::huggingface
+    config: {}
   - provider_id: localfs
     provider_type: inline::localfs
     config: {}