diff --git a/chat/config.py b/chat/config.py
index fcdc569..f2c19b8 100644
--- a/chat/config.py
+++ b/chat/config.py
@@ -65,6 +65,9 @@ class DataArguments:
     dataset_name: Optional[str] = field(
         default=None, metadata={"help": "The name of the dataset to use (via the datasets library)."}
     )
+    dataset_config_name: Optional[str] = field(
+        default=None, metadata={"help": "The configuration name of the dataset to use (via the datasets library)."}
+    )
     max_train_samples: Optional[int] = field(
         default=None,
         metadata={