Update README.md
Browse files
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-
|
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 |
|
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-
|
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]
|