Add GKD to dataset_formats.mdx (#2222)

* Update dataset_formats.mdx * Update dataset_formats.mdx * Update docs/source/dataset_formats.mdx Co-authored-by: Quentin Gallouédec <[email protected]> * Modified to Prompt-completion --------- Co-authored-by: Quentin Gallouédec <[email protected]>
huggingface · Oct 11, 2024 · b81a612 · b81a612
1 parent 7f0d246
commit b81a612
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/source/dataset_formats.mdx b/docs/source/dataset_formats.mdx
@@ -200,6 +200,7 @@ Choosing the right dataset format depends on the task you are working on and the
 | [`BCOTrainer`]          | [Unpaired preference](#unpaired-preference)             |
 | [`CPOTrainer`]          | [Preference (explicit prompt recommended)](#preference) |
 | [`DPOTrainer`]          | [Preference (explicit prompt recommended)](#preference) |
+| [`GKDTrainer`]          | [Prompt-completion](#prompt-completion)                 |
 | [`IterativeSFTTrainer`] | [Unpaired preference](#unpaired-preference)             |
 | [`KTOTrainer`]          | [Unpaired preference](#unpaired-preference)             |
 | [`NashMDTrainer`]       | [Prompt-only](#prompt-only)                             |