chaojiemao commited on
Commit
37a0d44
·
verified ·
1 Parent(s): 8491b63

Update config/models/ace_0.6b_1024.yaml

Browse files
Files changed (1) hide show
  1. config/models/ace_0.6b_1024.yaml +4 -4
config/models/ace_0.6b_1024.yaml CHANGED
@@ -70,7 +70,7 @@ MODEL:
70
  #
71
  DIFFUSION_MODEL:
72
  NAME: ACE
73
- PRETRAINED_MODEL: hf://iic/ACE-0.6B-1024px@models/dit/ace_0.6b_1024px.pth
74
  IGNORE_KEYS: [ ]
75
  PATCH_SIZE: 2
76
  IN_CHANNELS: 4
@@ -90,7 +90,7 @@ MODEL:
90
  FIRST_STAGE_MODEL:
91
  NAME: AutoencoderKL
92
  EMBED_DIM: 4
93
- PRETRAINED_MODEL: hf://iic/ACE-0.6B-1024px@models/vae/vae.bin
94
  IGNORE_KEYS: []
95
  #
96
  ENCODER:
@@ -122,8 +122,8 @@ MODEL:
122
  #
123
  COND_STAGE_MODEL:
124
  NAME: T5EmbedderHF
125
- PRETRAINED_MODEL: hf://iic/ACE-0.6B-1024px@models/text_encoder/t5-v1_1-xxl/
126
- TOKENIZER_PATH: hf://iic/ACE-0.6B-1024px@models/tokenizer/t5-v1_1-xxl
127
  LENGTH: 120
128
  T5_DTYPE: bfloat16
129
  ADDED_IDENTIFIER: [ '{image}', '{caption}', '{mask}', '{ref_image}', '{image1}', '{image2}', '{image3}', '{image4}', '{image5}', '{image6}', '{image7}', '{image8}', '{image9}' ]
 
70
  #
71
  DIFFUSION_MODEL:
72
  NAME: ACE
73
+ PRETRAINED_MODEL: hf://scepter-studio/ACE-0.6B-1024px@models/dit/ace_0.6b_1024px.pth
74
  IGNORE_KEYS: [ ]
75
  PATCH_SIZE: 2
76
  IN_CHANNELS: 4
 
90
  FIRST_STAGE_MODEL:
91
  NAME: AutoencoderKL
92
  EMBED_DIM: 4
93
+ PRETRAINED_MODEL: hf://scepter-studio/ACE-0.6B-1024px@models/vae/vae.bin
94
  IGNORE_KEYS: []
95
  #
96
  ENCODER:
 
122
  #
123
  COND_STAGE_MODEL:
124
  NAME: T5EmbedderHF
125
+ PRETRAINED_MODEL: hf://scepter-studio/ACE-0.6B-1024px@models/text_encoder/t5-v1_1-xxl/
126
+ TOKENIZER_PATH: hf://scepter-studio/ACE-0.6B-1024px@models/tokenizer/t5-v1_1-xxl
127
  LENGTH: 120
128
  T5_DTYPE: bfloat16
129
  ADDED_IDENTIFIER: [ '{image}', '{caption}', '{mask}', '{ref_image}', '{image1}', '{image2}', '{image3}', '{image4}', '{image5}', '{image6}', '{image7}', '{image8}', '{image9}' ]