Hugging Face's logo

Spaces:
scepter-studio
/
ACE-Chat
Running on Zero

App Files Community
1
chaojiemao commited on
Commit
5fbb38c
•
1 Parent(s): c0a5d5a

Update config/models/ace_0.6b_512.yaml

Browse files
Files changed (1) hide show
  1. config/models/ace_0.6b_512.yaml +5 -5
config/models/ace_0.6b_512.yaml CHANGED
@@ -39,12 +39,12 @@ DEFAULT_PARAS:
39
  #
40
  COND_STAGE_MODEL:
41
  FUNCTION:
42
- - NAME: encode_list
43
  DTYPE: bfloat16
44
  INPUT: ["PROMPT"]
45
  #
46
  MODEL:
47
- NAME: LdmACE
48
  PRETRAINED_MODEL:
49
  IGNORE_KEYS: [ ]
50
  SCALE_FACTOR: 0.18215
@@ -55,7 +55,7 @@ MODEL:
55
  USE_TEXT_POS_EMBEDDINGS: True
56
  #
57
  DIFFUSION:
58
- NAME: ACEDiffusion
59
  PREDICTION_TYPE: eps
60
  MIN_SNR_GAMMA:
61
  NOISE_SCHEDULER:
@@ -65,7 +65,7 @@ MODEL:
65
  BETA_MAX: 0.02
66
  #
67
  DIFFUSION_MODEL:
68
- NAME: DiTACE
69
  PRETRAINED_MODEL: hf://scepter-studio/ACE-0.6B-512px@models/dit/ace_0.6b_512px.pth
70
  IGNORE_KEYS: [ ]
71
  PATCH_SIZE: 2
@@ -117,7 +117,7 @@ MODEL:
117
  TANH_OUT: False
118
  #
119
  COND_STAGE_MODEL:
120
- NAME: ACETextEmbedder
121
  PRETRAINED_MODEL: hf://scepter-studio/ACE-0.6B-512px@models/text_encoder/t5-v1_1-xxl/
122
  TOKENIZER_PATH: hf://scepter-studio/ACE-0.6B-512px@models/tokenizer/t5-v1_1-xxl
123
  LENGTH: 120
 
39
  #
40
  COND_STAGE_MODEL:
41
  FUNCTION:
42
+ - NAME: encode_list_of_list
43
  DTYPE: bfloat16
44
  INPUT: ["PROMPT"]
45
  #
46
  MODEL:
47
+ NAME: LatentDiffusionACE
48
  PRETRAINED_MODEL:
49
  IGNORE_KEYS: [ ]
50
  SCALE_FACTOR: 0.18215
 
55
  USE_TEXT_POS_EMBEDDINGS: True
56
  #
57
  DIFFUSION:
58
+ NAME: BaseDiffusion
59
  PREDICTION_TYPE: eps
60
  MIN_SNR_GAMMA:
61
  NOISE_SCHEDULER:
 
65
  BETA_MAX: 0.02
66
  #
67
  DIFFUSION_MODEL:
68
+ NAME: ACE
69
  PRETRAINED_MODEL: hf://scepter-studio/ACE-0.6B-512px@models/dit/ace_0.6b_512px.pth
70
  IGNORE_KEYS: [ ]
71
  PATCH_SIZE: 2
 
117
  TANH_OUT: False
118
  #
119
  COND_STAGE_MODEL:
120
+ NAME: T5EmbedderHF
121
  PRETRAINED_MODEL: hf://scepter-studio/ACE-0.6B-512px@models/text_encoder/t5-v1_1-xxl/
122
  TOKENIZER_PATH: hf://scepter-studio/ACE-0.6B-512px@models/tokenizer/t5-v1_1-xxl
123
  LENGTH: 120