You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+16-17Lines changed: 16 additions & 17 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -6,23 +6,22 @@ NOTE: AutoTrain is free! You only pay for the resources you use in case you deci
6
6
7
7
## Supported Tasks
8
8
9
-
| Task | Status | Python Notebook |
10
-
| --- | --- | --- |
11
-
| LLM SFT Finetuning | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/llm_finetuning.ipynb)|
12
-
| LLM ORPO Finetuning | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/llm_finetuning.ipynb)|
13
-
| LLM DPO Finetuning | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/llm_finetuning.ipynb)|
14
-
| LLM Reward Finetuning | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/llm_finetuning.ipynb)|
15
-
| LLM Generic/Default Finetuning | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/llm_finetuning.ipynb)|
16
-
| Text Classification | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/text_classification.ipynb)|
17
-
| Text Regression | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/text_regression.ipynb)|
18
-
| Named Entity Recognition | β | Coming Soon |
19
-
| Token Classification | β | Coming Soon |
20
-
| Text Summarization | β | Coming Soon |
21
-
| Question Answering | β | Coming Soon |
22
-
| Image Classification | β | Coming Soon |
23
-
| Image Regression | β | Coming Soon |
24
-
| DreamBooth LoRA | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/colabs/AutoTrain_Dreambooth.ipynb)|
25
-
| VLM | π₯ | Coming Soon |
9
+
| Task | Status | Python Notebook | Example Configs |
10
+
| --- | --- | --- | --- |
11
+
| LLM SFT Finetuning | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/llm_finetuning.ipynb)|[llm_sft_finetune.yaml](https://github.com/huggingface/autotrain-advanced/blob/main/configs/llm_finetuning/smollm2.yml)|
12
+
| LLM ORPO Finetuning | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/llm_finetuning.ipynb)|[llm_orpo_finetune.yaml](https://github.com/huggingface/autotrain-advanced/blob/main/configs/llm_finetuning/llama3-8b-orpo.yml)|
13
+
| LLM DPO Finetuning | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/llm_finetuning.ipynb)|[llm_dpo_finetune.yaml](https://github.com/huggingface/autotrain-advanced/blob/main/configs/llm_finetuning/llama3-8b-dpo-qlora.yml)|
14
+
| LLM Reward Finetuning | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/llm_finetuning.ipynb)|[llm_reward_finetune.yaml](https://github.com/huggingface/autotrain-advanced/blob/main/configs/llm_finetuning/llama32-1b-sft.yml)|
15
+
| LLM Generic/Default Finetuning | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/llm_finetuning.ipynb)|[llm_generic_finetune.yaml](https://github.com/huggingface/autotrain-advanced/blob/main/configs/llm_finetuning/gpt2_sft.yml)|
16
+
| Text Classification | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/text_classification.ipynb)|[text_classification.yaml](https://github.com/huggingface/autotrain-advanced/tree/main/configs/text_classification)|
17
+
| Text Regression | β |[](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/notebooks/text_regression.ipynb)|[text_regression.yaml](https://github.com/huggingface/autotrain-advanced/tree/main/configs/text_regression)|
0 commit comments