Artiprocher commited on
Commit
e872915
·
1 Parent(s): 914a1c6

update model

Browse files
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.14.0.dev0",
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.14.0.dev0",
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.14.0.dev0",
4
- "_name_or_path": "models/20230306_512/checkpoint-120000/unet",
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:b2112036cac7e7cc226b63ae9dd42ce93458ea09161ad8db855378489619e189
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.14.0.dev0",
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": [