File size: 902 Bytes
55b9ef8 fbd6204 55b9ef8 a155279 55b9ef8 a155279 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
---
license: cc-by-sa-4.0
datasets:
- foldl/rumeme-desc
language:
- ru
library_name: diffusers
pipeline_tag: text-to-image
---
# ruMeme Description model
## Example
*Use `Diffusers` >=0.8.0, do not support lower versions.*
```python
from diffusers import StableDiffusionPipeline
import torch
model_path = "foldl/sd-rumeme-desc"
pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
pipe.to("cuda")
image = pipe(prompt="кот").images[0]
image.save("cat".jpg)
```
## Training
### Training Procedure
Model was trained on 1 P100 GPU for 10k steps.
Base model - https://huggingface.co/OFA-Sys/small-stable-diffusion-v0
Training notebook here - https://www.kaggle.com/code/nukeee/meme-diffusion
### Training Data
- https://huggingface.co/datasets/foldl/rumeme-desc
- Same dataset at [Kaggle](https://www.kaggle.com/datasets/nukeee/rumemes-descriptions) |