Update README.md

README.md

@@ -1,3 +1,40 @@
 ---
 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-Chinese-v0.1", torch_dtype=torch.float16)
+model_path = "svjack/pokemon-sd-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
+```