|
--- |
|
base_model: runwayml/stable-diffusion-v1-5 |
|
library_name: diffusers |
|
license: creativeml-openrail-m |
|
tags: |
|
- stable-diffusion |
|
- stable-diffusion-diffusers |
|
- text-to-image |
|
- diffusers |
|
- diffusers-training |
|
- lora |
|
inference: true |
|
--- |
|
|
|
<!-- This model card has been generated automatically according to the information the training script had access to. You |
|
should probably proofread and complete it, then remove this comment. --> |
|
|
|
|
|
# LoRA text2image fine-tuning - yashvoladoddi37/kanji-diffusion-v1-5 |
|
These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were fine-tuned on the yashvoladoddi37/kanjienglish dataset. You can find some example images in the following. |
|
|
|
![img_0](./image_0.png) |
|
![img_1](./image_1.png) |
|
![img_2](./image_2.png) |
|
![img_3](./image_3.png) |
|
|
|
#### How to use |
|
|
|
```python |
|
from diffusers import StableDiffusionPipeline |
|
import torch |
|
|
|
model_path = "yashvoladoddi37/kanji-diffusion-v1-5" |
|
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, use_safetensors = True).to("cuda") |
|
pipe.unet.load_attn_procs(model_path) |
|
pipe.to("cuda") |
|
|
|
prompt = "a Kanji meaning YouTube" |
|
image = pipe(prompt).images[0] |
|
image.save("youtube-kanji-v1-4.png") |
|
``` |