|
--- |
|
license: apache-2.0 |
|
datasets: |
|
- svjack/pokemon-blip-captions-en-zh |
|
language: |
|
- zh |
|
library_name: diffusers |
|
tags: |
|
- stable-diffusion |
|
- lora |
|
--- |
|
# Model Card for souljoy/sd-pokemon-model-lora-zh |
|
|
|
## Installation |
|
```bash |
|
pip install -U diffusers |
|
pip install transformers |
|
``` |
|
|
|
## Usage |
|
```python |
|
from diffusers import StableDiffusionPipeline |
|
import torch |
|
|
|
pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Anime-Chinese-v0.1", torch_dtype=torch.float16) |
|
model_path = "souljoy/sd-pokemon-model-lora-zh" |
|
|
|
pipe.unet.load_attn_procs(model_path) |
|
pipe.to("cuda") |
|
pipe.safety_checker = lambda images, clip_input: (images, False) |
|
print("have_load") |
|
|
|
prompt = "粉色的蝴蝶。" |
|
image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0] |
|
image |
|
|
|
prompt = "黄色的巨石机器人" |
|
image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0] |
|
image |
|
``` |
