Create README.md

README.md

@@ -0,0 +1,48 @@
+---
+datasets:
+- svjack/pokemon-blip-captions-en-zh
+language:
+- zh
+library_name: diffusers
+tags:
+- stable-diffusion
+- lora
+---
+# Model Card for svjack/pokemon-sd-lora-zh
+
+## Installation 
+```bash
+pip install -U diffusers
+pip install transformers
+```
+
+## Usage 
+```python
+from diffusers import StableDiffusionPipeline
+import torch
+
+'''
+pipe = StableDiffusionPipeline.from_pretrained("stable-diffusion-v1-2", torch_dtype=torch.float16)
+model_path = "en_lora_model"
+'''
+
+pipe = StableDiffusionPipeline.from_pretrained("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
+```
+
+![0](https://raw.githubusercontent.com/svjack/ControlLoRA-Chinese/main/imgs/zh_pink_butterfly_pokemon.png)
+![1](https://raw.githubusercontent.com/svjack/ControlLoRA-Chinese/main/imgs/zh_yellow_stone_robot.png)
+