ekhatskevich commited on
Commit
f28659f
·
1 Parent(s): fc8037f

fix config

Browse files
Files changed (1) hide show
  1. 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