Update README.md
Browse files
README.md
CHANGED
|
@@ -11,6 +11,8 @@ base_model: Salesforce/blip-image-captioning-base
|
|
| 11 |
# Overview
|
| 12 |
`ifmain/blip-image2prompt-stable-diffusion` is a model based on `Salesforce/blip-image-captioning-base`, trained on the `Ar4ikov/civitai-sd-337k` dataset (2K images). This model is designed to generate text descriptions of images in the style of prompts for use with Stable Diffusion models.
|
| 13 |
|
|
|
|
|
|
|
| 14 |
# Example Usage
|
| 15 |
```python
|
| 16 |
import torch
|
|
|
|
| 11 |
# Overview
|
| 12 |
`ifmain/blip-image2prompt-stable-diffusion` is a model based on `Salesforce/blip-image-captioning-base`, trained on the `Ar4ikov/civitai-sd-337k` dataset (2K images). This model is designed to generate text descriptions of images in the style of prompts for use with Stable Diffusion models.
|
| 13 |
|
| 14 |
+
I used my Blip training code: [BLIP-Easy-Trainer](https://github.com/ifmain/BLIP-Easy-Trainer)
|
| 15 |
+
|
| 16 |
# Example Usage
|
| 17 |
```python
|
| 18 |
import torch
|