Spaces:
Paused
Paused
ekhatskevich
commited on
Commit
·
f28659f
1
Parent(s):
fc8037f
fix config
Browse files- models/model_zoo.yaml +7 -1
models/model_zoo.yaml
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
MODEL:
|
2 |
PORTRAIT:
|
3 |
MODEL_PATH: ${PORTRAIT_MODEL_PATH}
|
@@ -25,4 +26,9 @@ MODEL:
|
|
25 |
PRETRAINED_MODEL: "ms://iic/scepter_annotator@annotator/ckpts/dpt_hybrid-midas-501f0c75.pt"
|
26 |
TYPE: depth_repainting
|
27 |
- NAME: GrayAnnotator
|
28 |
-
TYPE: recolorizing
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
NAME: FaceSwapConfig
|
2 |
MODEL:
|
3 |
PORTRAIT:
|
4 |
MODEL_PATH: ${PORTRAIT_MODEL_PATH}
|
|
|
26 |
PRETRAINED_MODEL: "ms://iic/scepter_annotator@annotator/ckpts/dpt_hybrid-midas-501f0c75.pt"
|
27 |
TYPE: depth_repainting
|
28 |
- NAME: GrayAnnotator
|
29 |
+
TYPE: recolorizing
|
30 |
+
MAX_SEQ_LEN: 77,
|
31 |
+
SAMPLE_ARGS: {
|
32 |
+
prompt: "Face swap"
|
33 |
+
}
|
34 |
+
DTYPE: bfloat16
|