remove dreambooth (#816)

Author: abhishekkrthakur

README.md

@@ -20,7 +20,6 @@ NOTE: AutoTrain is free! You only pay for the resources you use in case you deci
 | Extractive Question Answering | ✅ | Coming Soon | [extractive_qa.yaml](https://github.com/huggingface/autotrain-advanced/tree/main/configs/extractive_question_answering) |
 | Image Classification | ✅ | Coming Soon | [image_classification.yaml](https://github.com/huggingface/autotrain-advanced/tree/main/configs/image_classification) |
 | Image Scoring/Regression | ✅ | Coming Soon | [image_regression.yaml](https://github.com/huggingface/autotrain-advanced/tree/main/configs/image_scoring) |
-| DreamBooth LoRA | ✅ | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/huggingface/autotrain-advanced/blob/main/colabs/AutoTrain_Dreambooth.ipynb) | [dreambooth_lora.yaml](https://github.com/huggingface/autotrain-advanced/tree/main/configs/dreambooth) |
 | VLM | 🟥 | Coming Soon | [vlm.yaml](https://github.com/huggingface/autotrain-advanced/tree/main/configs/vlm) |
 
 

colabs/AutoTrain_Dreambooth.ipynb

@@ -31,8 +31,6 @@
     title: Image Classification / Regression
   - local: tasks/object_detection
     title: Object Detection
-  - local: tasks/dreambooth
-    title: DreamBooth
   - local: tasks/seq2seq
     title: Seq2Seq
   - local: tasks/token_classification

configs/dreambooth/sd15_colab.yml

@@ -131,10 +131,6 @@ For a single target column, you can pass a list with a single element.
 For multiple target columns, e.g. a multi label classification task, you can pass a list with multiple elements.
 
 
-# DreamBooth LoRA
-
-Dreambooth doesn't require column mapping.
-
 # Image Classification
 
 For image classification, the column mapping should be as follows:

configs/dreambooth/sdxl_colab.yml

@@ -55,7 +55,6 @@ positional arguments:
     app,
     llm,
     setup,
-    dreambooth,
     api,
     text-classification,
     text-regression,
@@ -82,7 +81,6 @@ The autotrain commands that end users will be interested in are:
 
 - `app`: Start the AutoTrain UI
 - `llm`: Train a language model
-- `dreambooth`: Train a model using DreamBooth
 - `text-classification`: Train a text classification model
 - `text-regression`: Train a text regression model
 - `image-classification`: Train an image classification model

docs/source/_toctree.yml

@@ -104,7 +104,6 @@ Your HF_TOKEN and HF_USERNAME are only required if you want to push the model or
 
 [[autodoc]] trainers.object_detection.params.ObjectDetectionParams
 
-[[autodoc]] trainers.dreambooth.params.DreamBoothTrainingParams
 
 ### Tabular Tasks
 

docs/source/col_map.mdx

@@ -18,7 +18,6 @@ xgboost==2.1.2
 huggingface_hub==0.26.2
 requests==2.32.3
 einops==0.8.0
-invisible-watermark==0.2.0
 packaging==24.1
 cryptography==43.0.3
 nvitop==1.3.2
@@ -29,7 +28,6 @@ trl==0.12.0
 tiktoken==0.6.0
 transformers==4.46.2
 accelerate==1.1.1
-diffusers==0.31.0
 bitsandbytes==0.44.1
 # extras
 rouge_score==0.1.2