Commit
·
e872915
1
Parent(s):
914a1c6
update model
Browse files- README.md +12 -0
- model_index.json +1 -1
- scheduler/scheduler_config.json +1 -1
- unet/config.json +2 -2
- unet/diffusion_pytorch_model.bin +1 -1
- vae/config.json +1 -1
README.md
CHANGED
@@ -11,3 +11,15 @@ tags:
|
|
11 |
我们开源了一个中文 Lattent Diffusion 模型(开放领域通用版本)
|
12 |
|
13 |
* Github: [EasyNLP](https://github.com/alibaba/EasyNLP)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
我们开源了一个中文 Lattent Diffusion 模型(开放领域通用版本)
|
12 |
|
13 |
* Github: [EasyNLP](https://github.com/alibaba/EasyNLP)
|
14 |
+
|
15 |
+
```python
|
16 |
+
from diffusers import StableDiffusionPipeline
|
17 |
+
|
18 |
+
model_id = "alibaba-pai/pai-diffusion-food-large-zh"
|
19 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id)
|
20 |
+
pipe = pipe.to("cuda")
|
21 |
+
|
22 |
+
prompt = "雾蒙蒙的日出在湖面上"
|
23 |
+
image = pipe(prompt).images[0]
|
24 |
+
image.save("result.png")
|
25 |
+
```
|
model_index.json
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
{
|
2 |
"_class_name": "StableDiffusionPipeline",
|
3 |
-
"_diffusers_version": "0.
|
4 |
"feature_extractor": [
|
5 |
"transformers",
|
6 |
"CLIPImageProcessor"
|
|
|
1 |
{
|
2 |
"_class_name": "StableDiffusionPipeline",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
"feature_extractor": [
|
5 |
"transformers",
|
6 |
"CLIPImageProcessor"
|
scheduler/scheduler_config.json
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
{
|
2 |
"_class_name": "DPMSolverMultistepScheduler",
|
3 |
-
"_diffusers_version": "0.
|
4 |
"algorithm_type": "dpmsolver++",
|
5 |
"beta_end": 0.012,
|
6 |
"beta_schedule": "scaled_linear",
|
|
|
1 |
{
|
2 |
"_class_name": "DPMSolverMultistepScheduler",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
"algorithm_type": "dpmsolver++",
|
5 |
"beta_end": 0.012,
|
6 |
"beta_schedule": "scaled_linear",
|
unet/config.json
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
{
|
2 |
"_class_name": "UNet2DConditionModel",
|
3 |
-
"_diffusers_version": "0.
|
4 |
-
"_name_or_path": "models/20230306_512/checkpoint-
|
5 |
"act_fn": "silu",
|
6 |
"attention_head_dim": 8,
|
7 |
"block_out_channels": [
|
|
|
1 |
{
|
2 |
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "models/20230306_512/checkpoint-125000/unet",
|
5 |
"act_fn": "silu",
|
6 |
"attention_head_dim": 8,
|
7 |
"block_out_channels": [
|
unet/diffusion_pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 3438354725
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ef14f74488a96566db1bd14f394058af980f52669fdff4350a06670aa84c5730
|
3 |
size 3438354725
|
vae/config.json
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
{
|
2 |
"_class_name": "AutoencoderKL",
|
3 |
-
"_diffusers_version": "0.
|
4 |
"_name_or_path": "models/sdm1.4_with_ChTextEncoder/vae",
|
5 |
"act_fn": "silu",
|
6 |
"block_out_channels": [
|
|
|
1 |
{
|
2 |
"_class_name": "AutoencoderKL",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
"_name_or_path": "models/sdm1.4_with_ChTextEncoder/vae",
|
5 |
"act_fn": "silu",
|
6 |
"block_out_channels": [
|