huggingface · abhishekkrthakur · Nov 29, 2024 · Nov 29, 2024 · Nov 29, 2024 · Nov 29, 2024
diff --git a/README.md b/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) |
 
 

diff --git a/colabs/AutoTrain_Dreambooth.ipynb b/colabs/AutoTrain_Dreambooth.ipynb
diff --git a/configs/dreambooth/sd15_colab.yml b/configs/dreambooth/sd15_colab.yml
diff --git a/configs/dreambooth/sdxl_colab.yml b/configs/dreambooth/sdxl_colab.yml
diff --git a/docs/source/_toctree.yml b/docs/source/_toctree.yml
@@ -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

diff --git a/docs/source/col_map.mdx b/docs/source/col_map.mdx
@@ -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:

diff --git a/docs/source/quickstart.mdx b/docs/source/quickstart.mdx
@@ -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

diff --git a/docs/source/quickstart_py.mdx b/docs/source/quickstart_py.mdx
@@ -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
 

diff --git a/docs/source/tasks/dreambooth.mdx b/docs/source/tasks/dreambooth.mdx
diff --git a/requirements.txt b/requirements.txt
@@ -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
Original file line number	Diff line number	Diff line change
Expand Up		@@ -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

Expand Down