Spaces:
Running
on
Zero
Running
on
Zero
Update config/models/ace_flux_dev.yaml
Browse files
config/models/ace_flux_dev.yaml
CHANGED
@@ -119,7 +119,7 @@ MODEL:
|
|
119 |
FIRST_STAGE_MODEL:
|
120 |
NAME: AutoencoderKLFlux
|
121 |
EMBED_DIM: 16
|
122 |
-
PRETRAINED_MODEL:
|
123 |
IGNORE_KEYS: [ ]
|
124 |
BATCH_SIZE: 8
|
125 |
USE_CONV: False
|
@@ -165,11 +165,11 @@ MODEL:
|
|
165 |
# HF_MODEL_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
166 |
HF_MODEL_CLS: T5EncoderModel
|
167 |
# MODEL_PATH DESCRIPTION: model folder path TYPE: NoneType default: None
|
168 |
-
MODEL_PATH:
|
169 |
# HF_TOKENIZER_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
170 |
HF_TOKENIZER_CLS: T5Tokenizer
|
171 |
# TOKENIZER_PATH DESCRIPTION: tokenizer folder path TYPE: NoneType default: None
|
172 |
-
TOKENIZER_PATH:
|
173 |
ADDED_IDENTIFIER: [ '<img>','{image}', '{caption}', '{mask}', '{ref_image}', '{image1}', '{image2}', '{image3}', '{image4}', '{image5}', '{image6}', '{image7}', '{image8}', '{image9}' ]
|
174 |
# MAX_LENGTH DESCRIPTION: max length of input TYPE: int default: 77
|
175 |
MAX_LENGTH: 512
|
@@ -187,11 +187,11 @@ MODEL:
|
|
187 |
# HF_MODEL_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
188 |
HF_MODEL_CLS: CLIPTextModel
|
189 |
# MODEL_PATH DESCRIPTION: model folder path TYPE: NoneType default: None
|
190 |
-
MODEL_PATH:
|
191 |
# HF_TOKENIZER_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
192 |
HF_TOKENIZER_CLS: CLIPTokenizer
|
193 |
# TOKENIZER_PATH DESCRIPTION: tokenizer folder path TYPE: NoneType default: None
|
194 |
-
TOKENIZER_PATH:
|
195 |
# MAX_LENGTH DESCRIPTION: max length of input TYPE: int default: 77
|
196 |
MAX_LENGTH: 77
|
197 |
# OUTPUT_KEY DESCRIPTION: output key TYPE: str default: 'last_hidden_state'
|
|
|
119 |
FIRST_STAGE_MODEL:
|
120 |
NAME: AutoencoderKLFlux
|
121 |
EMBED_DIM: 16
|
122 |
+
PRETRAINED_MODEL: hf://black-forest-labs/[email protected]
|
123 |
IGNORE_KEYS: [ ]
|
124 |
BATCH_SIZE: 8
|
125 |
USE_CONV: False
|
|
|
165 |
# HF_MODEL_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
166 |
HF_MODEL_CLS: T5EncoderModel
|
167 |
# MODEL_PATH DESCRIPTION: model folder path TYPE: NoneType default: None
|
168 |
+
MODEL_PATH: hf://black-forest-labs/FLUX.1-dev@text_encoder_2/
|
169 |
# HF_TOKENIZER_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
170 |
HF_TOKENIZER_CLS: T5Tokenizer
|
171 |
# TOKENIZER_PATH DESCRIPTION: tokenizer folder path TYPE: NoneType default: None
|
172 |
+
TOKENIZER_PATH: hf://black-forest-labs/FLUX.1-dev@tokenizer_2/
|
173 |
ADDED_IDENTIFIER: [ '<img>','{image}', '{caption}', '{mask}', '{ref_image}', '{image1}', '{image2}', '{image3}', '{image4}', '{image5}', '{image6}', '{image7}', '{image8}', '{image9}' ]
|
174 |
# MAX_LENGTH DESCRIPTION: max length of input TYPE: int default: 77
|
175 |
MAX_LENGTH: 512
|
|
|
187 |
# HF_MODEL_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
188 |
HF_MODEL_CLS: CLIPTextModel
|
189 |
# MODEL_PATH DESCRIPTION: model folder path TYPE: NoneType default: None
|
190 |
+
MODEL_PATH: hf://black-forest-labs/FLUX.1-dev@text_encoder/
|
191 |
# HF_TOKENIZER_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
192 |
HF_TOKENIZER_CLS: CLIPTokenizer
|
193 |
# TOKENIZER_PATH DESCRIPTION: tokenizer folder path TYPE: NoneType default: None
|
194 |
+
TOKENIZER_PATH: hf://black-forest-labs/FLUX.1-dev@tokenizer/
|
195 |
# MAX_LENGTH DESCRIPTION: max length of input TYPE: int default: 77
|
196 |
MAX_LENGTH: 77
|
197 |
# OUTPUT_KEY DESCRIPTION: output key TYPE: str default: 'last_hidden_state'
|