xiayu commited on
Commit
c375645
1 Parent(s): 62bb155

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -23,7 +23,7 @@ extra_gated_fields:
23
  I have read the License and agree with its terms: checkbox
24
  ---
25
 
26
- # Taiyi-Stable-Diffusion-1B-Bilingual-v0.1
27
 
28
  - Github: [Fengshenbang-LM](https://github.com/IDEA-CCNL/Fengshenbang-LM)
29
  - Docs: [Fengshenbang-Docs](https://fengshenbang-doc.readthedocs.io/)
@@ -38,7 +38,7 @@ The first open source Chinese&English Bilingual Stable diffusion, which was trai
38
 
39
  | 需求 Demand | 任务 Task | 系列 Series | 模型 Model | 参数 Parameter | 额外 Extra |
40
  | :----: | :----: | :----: | :----: | :----: | :----: |
41
- | 特殊 Special | 多模态 Multimodal | 太乙 Taiyi | Stable Diffusion | 1B | Bilingual |
42
 
43
  ## 模型信息 Model Information
44
 
@@ -78,7 +78,7 @@ It takes 80 hours to train the first stage, 100 hours to train the second stage,
78
  ```py
79
  from diffusers import StableDiffusionPipeline
80
 
81
- pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Bilingual-v0.1").to("cuda")
82
 
83
  prompt = '小桥流水人家,van gogh style'
84
  image = pipe(prompt, guidance_scale=10).images[0]
@@ -94,7 +94,7 @@ image.save("小桥.png")
94
  # !pip install git+https://github.com/huggingface/accelerate
95
  from diffusers import StableDiffusionPipeline
96
 
97
- pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1", torch_dtype=torch.float16, device_map="auto")
98
 
99
  prompt = '小桥流水人家,van gogh style'
100
  image = pipe(prompt, guidance_scale=10.0).images[0]
 
23
  I have read the License and agree with its terms: checkbox
24
  ---
25
 
26
+ # Taiyi-Stable-Diffusion-1B-Chinese-EN-v0.1
27
 
28
  - Github: [Fengshenbang-LM](https://github.com/IDEA-CCNL/Fengshenbang-LM)
29
  - Docs: [Fengshenbang-Docs](https://fengshenbang-doc.readthedocs.io/)
 
38
 
39
  | 需求 Demand | 任务 Task | 系列 Series | 模型 Model | 参数 Parameter | 额外 Extra |
40
  | :----: | :----: | :----: | :----: | :----: | :----: |
41
+ | 特殊 Special | 多模态 Multimodal | 太乙 Taiyi | Stable Diffusion | 1B | Chinese and English |
42
 
43
  ## 模型信息 Model Information
44
 
 
78
  ```py
79
  from diffusers import StableDiffusionPipeline
80
 
81
+ pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-EN-v0.1").to("cuda")
82
 
83
  prompt = '小桥流水人家,van gogh style'
84
  image = pipe(prompt, guidance_scale=10).images[0]
 
94
  # !pip install git+https://github.com/huggingface/accelerate
95
  from diffusers import StableDiffusionPipeline
96
 
97
+ pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-EN-v0.1", torch_dtype=torch.float16, device_map="auto")
98
 
99
  prompt = '小桥流水人家,van gogh style'
100
  image = pipe(prompt, guidance_scale=10.0).images[0]