File size: 811 Bytes
5b3b410 | 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 | ---
tags:
- stable-diffusion
- lora
- text-to-image
- diffusers
base_model: stable-diffusion-v1-5/stable-diffusion-v1-5
license: creativeml-openrail-m
library_name: diffusers
pipeline_tag: text-to-image
---
# LoRA fine-tuned on custom dataset
## Details
- **Base model**: `stable-diffusion-v1-5/stable-diffusion-v1-5`
- **Resolution**: 512px
- **Method**: LoRA (rank 16)
- **Training steps**: 1500
- **Learning rate**: 0.0001
## Usage
```python
import torch
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained(
"stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float16
).to("cuda")
pipe.load_lora_weights("tenith/pokemon-lora")
image = pipe("your prompt here", num_inference_steps=30, guidance_scale=7.5).images[0]
image.save("output.png")
```
|