Commit
•
940bc46
1
Parent(s):
c9b67f3
Update README.MD
Browse files
README.md
CHANGED
|
@@ -6,4 +6,38 @@ tags:
|
|
| 6 |
base_model: stabilityai/stable-diffusion-xl-base-1.0
|
| 7 |
license: apache-2.0
|
| 8 |
pipeline_tag: text-to-image
|
| 9 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
base_model: stabilityai/stable-diffusion-xl-base-1.0
|
| 7 |
license: apache-2.0
|
| 8 |
pipeline_tag: text-to-image
|
| 9 |
+
---
|
| 10 |
+
# DreamBooth LoRA Training with Stable Diffusion XL on Trump Thumbs Up Images
|
| 11 |
+
|
| 12 |
+
This repository contains instructions and code for training a DreamBooth LoRA model using Stable Diffusion XL on a dataset of images featuring Donald Trump giving a thumbs up gesture. The trained model can be used to generate high-quality images of Trump showing thumbs up in various contexts.
|
| 13 |
+
|
| 14 |
+
## Sample Images
|
| 15 |
+
|
| 16 |
+
Here are a few sample images generated by the trained model:
|
| 17 |
+
|
| 18 |
+
|
| 19 |
+
*A high quality picture of Trump showing thumbs up in a busy street of India, detailed, sharp focus*
|
| 20 |
+
|
| 21 |
+
|
| 22 |
+
*An intricately detailed digital painting of Donald Trump giving a thumbs up at a taco restaurant. The background includes colorful decor and a bustling atmosphere with people enjoying their meals.*
|
| 23 |
+
|
| 24 |
+
|
| 25 |
+
*A high-quality photo of Donald Trump giving a thumbs up on a sunny beach. The scene includes clear blue water, white sand, and Trump in casual beachwear. The image is detailed, with Trump’s smiling face and the vibrant beach setting in sharp focus.*.
|
| 26 |
+
|
| 27 |
+
## Requirements
|
| 28 |
+
|
| 29 |
+
The script requires Python 3.9 and several Python packages including PyTorch, Hugging Face Transformers, Diffusers, and Accelerate. Additional dependencies are listed in the `requirements_sdxl.txt` file.
|
| 30 |
+
|
| 31 |
+
## Installation
|
| 32 |
+
|
| 33 |
+
To get started, clone the repository and navigate to the project directory. Install the required packages using pip and the provided `requirements_sdxl.txt` file. Log in to the Hugging Face Hub using the `huggingface-cli login` command.
|
| 34 |
+
|
| 35 |
+
## Usage
|
| 36 |
+
|
| 37 |
+
To train the model, prepare a dataset of images featuring Donald Trump giving a thumbs up gesture and place them in a directory. Run the training script `train_dreambooth_lora_sdxl.py` with the appropriate command-line arguments specifying the pretrained model, instance data directory, output directory, and various training hyperparameters.
|
| 38 |
+
|
| 39 |
+
After training, load the trained LoRA weights and use the `DiffusionPipeline` class from the Diffusers library to generate images. Provide a prompt describing the desired image, such as "A high quality picture of Trump showing the thumbs up in Paris detailed, sharp focus". The generated image will be saved to the specified output directory.
|
| 40 |
+
|
| 41 |
+
## Results
|
| 42 |
+
|
| 43 |
+
The generated images will be saved in the specified output directory, showcasing Trump giving a thumbs up gesture in different contexts based on the provided prompts.
|