ramstorage / ai-avatars

This repository exploits a finely tuned Stable Diffusion model and the robust DreamBooth framework to generate distinctive and personalizable avatars.

AI Personalized Avatars

This project harnesses the power of artificial intelligence to generate unique, personalized avatars based on a set of training images. The model used in this project is a variant of the Stable Diffusion model, known for its exceptional capabilities in image generation tasks.

Sample Images

Here are some example input training images and corresponding output images generated by the AI model:

Training Images:

Generated Avatars:

Avatar 1 - Style: Historic and Gaming Character


Avatar 2 - Style: Abstract Art


Avatar 3 - Style: Comic and Cyberpunk

Repository Contents

This repository contains:

1. Google Colab Notebook: A pre-configured Google Colab notebook for training your model. The notebook includes steps for setting up the environment, loading the necessary pre-trained models, uploading training images, training the model, and saving the model checkpoints.

2. Regularization Images: A set of regularization images used for stabilizing the training process. You can add more images or replace these images as per your requirements.

3. Training Images Directory: A dedicated directory to upload your training images. The images you upload will be used to personalize the AI model, and should be representative of the avatars you wish to generate.

4. Trained Models: The final trained model checkpoints saved in a directory named "trained_models".

How to Use

Follow these steps to create your own AI personalized avatars:

1. Clone or Download this Repository: You can clone this repository to your local machine or simply download it.

2. Setup the Environment: Install the necessary Python packages which are listed in the Google Colab notebook.

3. Download the Pre-trained Model: The Google Colab notebook contains steps to download a pre-trained model from Hugging Face. This pre-trained model serves as a starting point for training your personalized avatar model.

4. Upload Training Images: You can upload your own set of training images in the "training_images" directory. These images will be used to fine-tune the pre-trained model. Ensure the images are representative of the avatars you wish the AI to generate.

5. Train the Model: Run the training script in the Google Colab notebook. This script fine-tunes the pre-trained model on your training images.

6. Save and Use the Trained Model: After training, the resulting model checkpoint is saved in the "trained_models" directory. You can now use this model checkpoint to generate your own personalized avatars.