chaojiemao commited on
Commit
3000b0a
·
verified ·
1 Parent(s): a2decc8

Update config/models/ace_flux_dev.yaml

Browse files
Files changed (1) hide show
  1. config/models/ace_flux_dev.yaml +7 -7
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: ms://AI-ModelScope/FLUX.1-dev@flux1-dev.safetensors
75
- SWIFT_LORA_MODEL: ["ms://iic/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,7 +103,7 @@ MODEL:
103
  FIRST_STAGE_MODEL:
104
  NAME: AutoencoderKLFlux
105
  EMBED_DIM: 16
106
- PRETRAINED_MODEL: ms://AI-ModelScope/FLUX.1-dev@ae.safetensors
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: ms://AI-ModelScope/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: ms://AI-ModelScope/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,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: ms://AI-ModelScope/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: ms://AI-ModelScope/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'
 
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'