From e6b11eb76435411cd81f2a281742a60ae9baaa8a Mon Sep 17 00:00:00 2001 From: GiganticPrime Date: Wed, 28 Jun 2023 06:13:10 +0000 Subject: [PATCH] add 'dataset_config_name' field to config.py with metadata of transformer --- chat/config.py | 3 +++ 1 file changed, 3 insertions(+) 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={