Spaces:
Runtime error
Runtime error
Update config/models/ace_0.6b_512.yaml
Browse files
config/models/ace_0.6b_512.yaml
CHANGED
|
@@ -39,12 +39,12 @@ DEFAULT_PARAS:
|
|
| 39 |
#
|
| 40 |
COND_STAGE_MODEL:
|
| 41 |
FUNCTION:
|
| 42 |
-
- NAME:
|
| 43 |
DTYPE: bfloat16
|
| 44 |
INPUT: ["PROMPT"]
|
| 45 |
#
|
| 46 |
MODEL:
|
| 47 |
-
NAME:
|
| 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:
|
| 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:
|
| 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:
|
| 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
|