zainikhan34 commited on
Commit
50dad72
·
verified ·
1 Parent(s): 882ee77

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -25
README.md CHANGED
@@ -4,41 +4,44 @@ tags:
4
  - text-to-image
5
  - stable-diffusion
6
  ---
7
- ### Fine-Tune-Stable-Diffusion-model-for-Generating-My-pics Dreambooth model trained by zainikhan34 with [TheLastBen's fast-DreamBooth](https://colab.research.google.com/github/TheLastBen/fast-stable-diffusion/blob/main/fast-DreamBooth.ipynb) notebook
8
 
 
9
 
10
- Fine-Tuned Stable Diffusion Model with Dreambooth
11
- Model Description
12
  This model is a fine-tuned version of the Stable Diffusion model using Dreambooth techniques. It is trained to generate high-quality images based on specific prompts, with a focus on generating images of the user.
13
 
14
- Training Data
15
- Dataset: The training dataset consists of 6 images of the user, Zain.
16
- Captions: All images were captioned with the user's name, "Zain".
17
- Preprocessing: Standard preprocessing steps were applied to the images.
18
- Training Procedure
19
- Training Method: Dreambooth
20
- UNet Training Steps: 1500
21
- UNet Learning Rate: 2e-6
22
- Resolution: 512
23
- Results
24
- Generated Images
25
- Here are some example images generated by the model:
26
 
 
 
 
 
 
27
 
 
 
 
28
 
 
 
29
 
30
- Intended Use
31
  This model is designed to generate personalized images based on the user's input. Potential applications include generating avatars, artistic portraits, and other creative outputs featuring the user.
32
 
33
- Limitations and Ethical Considerations
34
- Limitations: The model is not generating the user's face accurately, as the training images were taken from a distance and did not focus on the face.
35
- Ethical Considerations: Users should be aware of the ethical implications of using AI-generated images, particularly in terms of privacy and the potential for misuse.
36
- How to Use
37
- Instructions on how to use the model can be found in the Hugging Face documentation.
 
38
 
39
- Acknowledgments
40
- Thanks to TheLastBen's fast-DreamBooth notebook for providing the training framework.
41
- Contact
 
42
  For further questions or collaborations, please contact zainikhan3434@gmail.com.
43
- Sample pictures of this concept:
44
 
 
4
  - text-to-image
5
  - stable-diffusion
6
  ---
 
7
 
8
+ # Fine-Tuned Stable Diffusion Model with Dreambooth
9
 
10
+ ## Model Description
 
11
  This model is a fine-tuned version of the Stable Diffusion model using Dreambooth techniques. It is trained to generate high-quality images based on specific prompts, with a focus on generating images of the user.
12
 
13
+ ## Training Data
14
+ - **Dataset:** The training dataset consists of 6 images of the user, Zain.
15
+ - **Captions:** All images were captioned with the user's name, "Zain".
16
+ - **Preprocessing:** Standard preprocessing steps were applied to the images.
 
 
 
 
 
 
 
 
17
 
18
+ ## Training Procedure
19
+ - **Training Method:** Dreambooth
20
+ - **UNet Training Steps:** 1500
21
+ - **UNet Learning Rate:** 2e-6
22
+ - **Resolution:** 512
23
 
24
+ ## Results
25
+ ### Generated Images
26
+ Here are some example images generated by the model:
27
 
28
+ ![Generated Image 1](https://huggingface.co/zainikhan34/ZainVision_Fine-Tuned_Stable_Diffusion_Model_with_Dreambooth/blob/main/8b87f486-48dc-4411-b5b9-14a0d88dccb9.jpg)
29
+ ![Generated Image 2](https://huggingface.co/zainikhan34/ZainVision_Fine-Tuned_Stable_Diffusion_Model_with_Dreambooth/blob/main/8b87f486-48dc-4411-b5b9-14a0d88dccb9.jpg)
30
 
31
+ ## Intended Use
32
  This model is designed to generate personalized images based on the user's input. Potential applications include generating avatars, artistic portraits, and other creative outputs featuring the user.
33
 
34
+ ## Limitations and Ethical Considerations
35
+ - **Limitations:** The model is not generating the user's face accurately, as the training images were taken from a distance and did not focus on the face.
36
+ - **Ethical Considerations:** Users should be aware of the ethical implications of using AI-generated images, particularly in terms of privacy and the potential for misuse.
37
+
38
+ ## How to Use
39
+ Instructions on how to use the model can be found in the [Hugging Face documentation](https://huggingface.co/docs).
40
 
41
+ ## Acknowledgements
42
+ - Thanks to [TheLastBen's fast-DreamBooth notebook](https://github.com/TheLastBen/fast-DreamBooth) for providing the training framework.
43
+
44
+ ## Contact
45
  For further questions or collaborations, please contact zainikhan3434@gmail.com.
46
+
47