add configuration

app.py

@@ -11,18 +11,9 @@ os.environ["LOCAL_MODEL_PATH"] = "ms://iic/ACE_Plus@local_editing/comfyui_local_
 from inference.ace_plus_inference import ACEInference
 from scepter.modules.utils.config import Config
 
-config_dict = {
-    "MODEL": {
-        "type": "ACE",
-        "pretrained_path": os.getenv("PORTRAIT_MODEL_PATH")
-    },
-    "MAX_SEQ_LEN": 77,
-    "SAMPLE_ARGS": {
-        "prompt": "Face swap"
-    },
-    "DTYPE": "bfloat16"
-}
-cfg = Config(config_dict)
+
+config_path = os.path.join("models", "model_zoo.yaml")
+cfg = Config(load=True, cfg_file=config_path)
 
 # Instantiate the ACEInference object.
 ace_infer = ACEInference(cfg)

models/model_zoo.yaml

@@ -0,0 +1,28 @@
+MODEL:
+  PORTRAIT:
+    MODEL_PATH: ${PORTRAIT_MODEL_PATH}
+  SUBJECT:
+    MODEL_PATH: ${SUBJECT_MODEL_PATH}
+  LOCAL_EDITING:
+    MODEL_PATH: ${LOCAL_MODEL_PATH}
+    REPAINTING_SCALE: 0.5
+    PREPROCESSOR:
+      - NAME: CannyAnnotator
+        TYPE: canny_repaintingß
+        LOW_THRESHOLD: 100
+        HIGH_THRESHOLD: 200
+      - NAME: ColorAnnotator
+        TYPE: mosaic_repainting
+        RATIO: 64
+      - NAME: InfoDrawContourAnnotator
+        TYPE: contour_repainting
+        INPUT_NC: 3
+        OUTPUT_NC: 1
+        N_RESIDUAL_BLOCKS: 3
+        SIGMOID: True
+        PRETRAINED_MODEL: "ms://iic/scepter_annotator@annotator/ckpts/informative_drawing_contour_style.pth"
+      - NAME: MidasDetector
+        PRETRAINED_MODEL: "ms://iic/scepter_annotator@annotator/ckpts/dpt_hybrid-midas-501f0c75.pt"
+        TYPE: depth_repainting
+      - NAME: GrayAnnotator
+        TYPE: recolorizing