diff --git a/config/ace_plus_fft.yaml b/config/ace_plus_fft.yaml index 889c92d..8250630 100644 --- a/config/ace_plus_fft.yaml +++ b/config/ace_plus_fft.yaml @@ -72,7 +72,7 @@ MODEL: FIRST_STAGE_MODEL: NAME: AutoencoderKLFlux EMBED_DIM: 16 - PRETRAINED_MODEL: ${FLUX_FILL_PATH}/ae.safetensors + PRETRAINED_MODEL: ${FLUX_FILL_PATH}@ae.safetensors IGNORE_KEYS: [ ] BATCH_SIZE: 8 USE_CONV: False @@ -116,11 +116,11 @@ MODEL: # HF_MODEL_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None HF_MODEL_CLS: T5EncoderModel # MODEL_PATH DESCRIPTION: model folder path TYPE: NoneType default: None - MODEL_PATH: ${FLUX_FILL_PATH}/text_encoder_2/ + MODEL_PATH: ${FLUX_FILL_PATH}@text_encoder_2/ # HF_TOKENIZER_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None HF_TOKENIZER_CLS: T5Tokenizer # TOKENIZER_PATH DESCRIPTION: tokenizer folder path TYPE: NoneType default: None - TOKENIZER_PATH: ${FLUX_FILL_PATH}/tokenizer_2/ + TOKENIZER_PATH: ${FLUX_FILL_PATH}@tokenizer_2/ ADDED_IDENTIFIER: [ '','{image}', '{caption}', '{mask}', '{ref_image}', '{image1}', '{image2}', '{image3}', '{image4}', '{image5}', '{image6}', '{image7}', '{image8}', '{image9}' ] # MAX_LENGTH DESCRIPTION: max length of input TYPE: int default: 77 MAX_LENGTH: 512 @@ -138,11 +138,11 @@ MODEL: # HF_MODEL_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None HF_MODEL_CLS: CLIPTextModel # MODEL_PATH DESCRIPTION: model folder path TYPE: NoneType default: None - MODEL_PATH: ${FLUX_FILL_PATH}/text_encoder/ + MODEL_PATH: ${FLUX_FILL_PATH}@text_encoder/ # HF_TOKENIZER_CLS DESCRIPTION: huggingface cls in transfomer TYPE: NoneType default: None HF_TOKENIZER_CLS: CLIPTokenizer # TOKENIZER_PATH DESCRIPTION: tokenizer folder path TYPE: NoneType default: None - TOKENIZER_PATH: ${FLUX_FILL_PATH}/tokenizer/ + TOKENIZER_PATH: ${FLUX_FILL_PATH}@tokenizer/ # MAX_LENGTH DESCRIPTION: max length of input TYPE: int default: 77 MAX_LENGTH: 77 # OUTPUT_KEY DESCRIPTION: output key TYPE: str default: 'last_hidden_state'