Upload README.md

README.md

@@ -11,7 +11,7 @@ tags:
 
 
 
-# Midu-Stable-Diffusion-2-Chinese-Style-v0.1
+# Chinese-Style-Stable-Diffusion-2-v0.1
 
 
 
@@ -45,12 +45,23 @@ Training on 5M chinese style filtered data for 150k steps. Exponential moving av
 
 ## Usage
 
-因为使用了customed tokenizer, 所以需要优先加载一下tokenizer
+因为使用了customed tokenizer, 所以需要优先加载一下tokenizer, 并传入trust_remote_code=True
+
+Custom Tokenizer should be loaded first with 'trust_remote_code=True'.
 
 ```py
-# !pip install git+https://github.com/huggingface/accelerate
 import torch
 from diffusers import StableDiffusionPipeline
+from transformers import AutoTokenizer
+
+tokenizer_id = "lyua1225/clip-huge-zh-75k-steps-bs4096"
+sd2_id = "Midu/chinese-style-stable-diffusion-2-v0.1"
+tokenizer = AutoTokenizer.from_pretrained(tokenizer_id, trust_remote_code=True)
+pipe = StableDiffusionPipeline.from_pretrained(sd2_id, torch_dtype=torch.float16, tokenizer=tokenizer)
+pipe.to("cuda")
+
+image = pipe("赛博朋克风格的城市街道，8K分辨率，CG渲染", guidance_scale=10, num_inference_steps=20).images[0]
+image.save("cyberpunk.jpeg")
 
 ```