Spaces:
Running
on
Zero
Running
on
Zero
Update config/models/ace_flux_dev.yaml
Browse files
config/models/ace_flux_dev.yaml
CHANGED
@@ -71,8 +71,8 @@ MODEL:
|
|
71 |
DIFFUSION_MODEL:
|
72 |
# NAME DESCRIPTION: TYPE: default: 'Flux'
|
73 |
NAME: ACEFlux
|
74 |
-
PRETRAINED_MODEL:
|
75 |
-
SWIFT_LORA_MODEL: ["
|
76 |
# IN_CHANNELS DESCRIPTION: model's input channels. TYPE: int default: 64
|
77 |
IN_CHANNELS: 64
|
78 |
# HIDDEN_SIZE DESCRIPTION: model's hidden size. TYPE: int default: 1024
|
@@ -103,7 +103,7 @@ MODEL:
|
|
103 |
FIRST_STAGE_MODEL:
|
104 |
NAME: AutoencoderKLFlux
|
105 |
EMBED_DIM: 16
|
106 |
-
PRETRAINED_MODEL:
|
107 |
IGNORE_KEYS: [ ]
|
108 |
BATCH_SIZE: 8
|
109 |
USE_CONV: False
|
@@ -149,11 +149,11 @@ MODEL:
|
|
149 |
# HF_MODEL_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
150 |
HF_MODEL_CLS: T5EncoderModel
|
151 |
# MODEL_PATH DESCRIPTION: model folder path TYPE: NoneType default: None
|
152 |
-
MODEL_PATH:
|
153 |
# HF_TOKENIZER_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
154 |
HF_TOKENIZER_CLS: T5Tokenizer
|
155 |
# TOKENIZER_PATH DESCRIPTION: tokenizer folder path TYPE: NoneType default: None
|
156 |
-
TOKENIZER_PATH:
|
157 |
ADDED_IDENTIFIER: [ '<img>','{image}', '{caption}', '{mask}', '{ref_image}', '{image1}', '{image2}', '{image3}', '{image4}', '{image5}', '{image6}', '{image7}', '{image8}', '{image9}' ]
|
158 |
# MAX_LENGTH DESCRIPTION: max length of input TYPE: int default: 77
|
159 |
MAX_LENGTH: 512
|
@@ -171,11 +171,11 @@ MODEL:
|
|
171 |
# HF_MODEL_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
172 |
HF_MODEL_CLS: CLIPTextModel
|
173 |
# MODEL_PATH DESCRIPTION: model folder path TYPE: NoneType default: None
|
174 |
-
MODEL_PATH:
|
175 |
# HF_TOKENIZER_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
176 |
HF_TOKENIZER_CLS: CLIPTokenizer
|
177 |
# TOKENIZER_PATH DESCRIPTION: tokenizer folder path TYPE: NoneType default: None
|
178 |
-
TOKENIZER_PATH:
|
179 |
# MAX_LENGTH DESCRIPTION: max length of input TYPE: int default: 77
|
180 |
MAX_LENGTH: 77
|
181 |
# OUTPUT_KEY DESCRIPTION: output key TYPE: str default: 'last_hidden_state'
|
|
|
71 |
DIFFUSION_MODEL:
|
72 |
# NAME DESCRIPTION: TYPE: default: 'Flux'
|
73 |
NAME: ACEFlux
|
74 |
+
PRETRAINED_MODEL: hf://black-forest-labs/FLUX.1-dev@flux1-dev.safetensors
|
75 |
+
SWIFT_LORA_MODEL: ["hf://scepter-studio/ACE-FLUX.1-dev@ace_flux.1_dev_lora.bin"]
|
76 |
# IN_CHANNELS DESCRIPTION: model's input channels. TYPE: int default: 64
|
77 |
IN_CHANNELS: 64
|
78 |
# HIDDEN_SIZE DESCRIPTION: model's hidden size. TYPE: int default: 1024
|
|
|
103 |
FIRST_STAGE_MODEL:
|
104 |
NAME: AutoencoderKLFlux
|
105 |
EMBED_DIM: 16
|
106 |
+
PRETRAINED_MODEL: hf://black-forest-labs/FLUX.1-dev@ae.safetensors
|
107 |
IGNORE_KEYS: [ ]
|
108 |
BATCH_SIZE: 8
|
109 |
USE_CONV: False
|
|
|
149 |
# HF_MODEL_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
150 |
HF_MODEL_CLS: T5EncoderModel
|
151 |
# MODEL_PATH DESCRIPTION: model folder path TYPE: NoneType default: None
|
152 |
+
MODEL_PATH: hf://black-forest-labs/FLUX.1-dev@text_encoder_2/
|
153 |
# HF_TOKENIZER_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
154 |
HF_TOKENIZER_CLS: T5Tokenizer
|
155 |
# TOKENIZER_PATH DESCRIPTION: tokenizer folder path TYPE: NoneType default: None
|
156 |
+
TOKENIZER_PATH: hf://black-forest-labs/FLUX.1-dev@tokenizer_2/
|
157 |
ADDED_IDENTIFIER: [ '<img>','{image}', '{caption}', '{mask}', '{ref_image}', '{image1}', '{image2}', '{image3}', '{image4}', '{image5}', '{image6}', '{image7}', '{image8}', '{image9}' ]
|
158 |
# MAX_LENGTH DESCRIPTION: max length of input TYPE: int default: 77
|
159 |
MAX_LENGTH: 512
|
|
|
171 |
# HF_MODEL_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
172 |
HF_MODEL_CLS: CLIPTextModel
|
173 |
# MODEL_PATH DESCRIPTION: model folder path TYPE: NoneType default: None
|
174 |
+
MODEL_PATH: hf://black-forest-labs/FLUX.1-dev@text_encoder/
|
175 |
# HF_TOKENIZER_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None
|
176 |
HF_TOKENIZER_CLS: CLIPTokenizer
|
177 |
# TOKENIZER_PATH DESCRIPTION: tokenizer folder path TYPE: NoneType default: None
|
178 |
+
TOKENIZER_PATH: hf://black-forest-labs/FLUX.1-dev@tokenizer/
|
179 |
# MAX_LENGTH DESCRIPTION: max length of input TYPE: int default: 77
|
180 |
MAX_LENGTH: 77
|
181 |
# OUTPUT_KEY DESCRIPTION: output key TYPE: str default: 'last_hidden_state'
|