Update config/models/ace_0.6b_1024.yaml

config/models/ace_0.6b_1024.yaml

@@ -70,7 +70,7 @@ MODEL:
   #
   DIFFUSION_MODEL:
     NAME: ACE
-    PRETRAINED_MODEL: ms://iic/ACE-0.6B-1024px@models/dit/ace_0.6b_1024px.pth
+    PRETRAINED_MODEL: hf://iic/ACE-0.6B-1024px@models/dit/ace_0.6b_1024px.pth
     IGNORE_KEYS: [ ]
     PATCH_SIZE: 2
     IN_CHANNELS: 4
@@ -90,7 +90,7 @@ MODEL:
   FIRST_STAGE_MODEL:
     NAME: AutoencoderKL
     EMBED_DIM: 4
-    PRETRAINED_MODEL: ms://iic/ACE-0.6B-1024px@models/vae/vae.bin
+    PRETRAINED_MODEL: hf://iic/ACE-0.6B-1024px@models/vae/vae.bin
     IGNORE_KEYS: []
     #
     ENCODER:
@@ -122,8 +122,8 @@ MODEL:
   #
   COND_STAGE_MODEL:
     NAME: T5EmbedderHF
-    PRETRAINED_MODEL: ms://iic/ACE-0.6B-1024px@models/text_encoder/t5-v1_1-xxl/
-    TOKENIZER_PATH: ms://iic/ACE-0.6B-1024px@models/tokenizer/t5-v1_1-xxl
+    PRETRAINED_MODEL: hf://iic/ACE-0.6B-1024px@models/text_encoder/t5-v1_1-xxl/
+    TOKENIZER_PATH: hf://iic/ACE-0.6B-1024px@models/tokenizer/t5-v1_1-xxl
     LENGTH: 120
     T5_DTYPE: bfloat16
     ADDED_IDENTIFIER: [ '{image}', '{caption}', '{mask}', '{ref_image}', '{image1}', '{image2}', '{image3}', '{image4}', '{image5}', '{image6}', '{image7}', '{image8}', '{image9}' ]
@@ -201,7 +201,7 @@ REFINER_MODEL:
         MODE_SCALE: 1.29
     DIFFUSION_MODEL:
       NAME: FluxMR
-      PRETRAINED_MODEL: ms://AI-ModelScope/[email protected]
+      PRETRAINED_MODEL: hf://black-forest-labs/[email protected]
       IN_CHANNELS: 64
       OUT_CHANNELS: 64
       HIDDEN_SIZE: 3072
@@ -221,7 +221,7 @@ REFINER_MODEL:
     FIRST_STAGE_MODEL:
       NAME: AutoencoderKLFlux
       EMBED_DIM: 16
-      PRETRAINED_MODEL:  ms://AI-ModelScope/[email protected]
+      PRETRAINED_MODEL:  hf://black-forest-labs/[email protected]
       IGNORE_KEYS: [ ]
       BATCH_SIZE: 8
       USE_CONV: False
@@ -262,9 +262,9 @@ REFINER_MODEL:
       T5_MODEL:
         NAME: HFEmbedder
         HF_MODEL_CLS: T5EncoderModel
-        MODEL_PATH: ms://AI-ModelScope/FLUX.1-dev@text_encoder_2/
+        MODEL_PATH: hf://black-forest-labs/FLUX.1-dev@text_encoder_2/
         HF_TOKENIZER_CLS: T5Tokenizer
-        TOKENIZER_PATH: ms://AI-ModelScope/FLUX.1-dev@tokenizer_2/
+        TOKENIZER_PATH: hf://black-forest-labs/FLUX.1-dev@tokenizer_2/
         MAX_LENGTH: 512
         OUTPUT_KEY: last_hidden_state
         D_TYPE: bfloat16
@@ -273,9 +273,9 @@ REFINER_MODEL:
       CLIP_MODEL:
         NAME: HFEmbedder
         HF_MODEL_CLS: CLIPTextModel
-        MODEL_PATH: ms://AI-ModelScope/FLUX.1-dev@text_encoder/
+        MODEL_PATH: hf://black-forest-labs/FLUX.1-dev@text_encoder/
         HF_TOKENIZER_CLS: CLIPTokenizer
-        TOKENIZER_PATH: ms://AI-ModelScope/FLUX.1-dev@tokenizer/
+        TOKENIZER_PATH: hf://black-forest-labs/FLUX.1-dev@tokenizer/
         MAX_LENGTH: 77
         OUTPUT_KEY: pooler_output
         D_TYPE: bfloat16