WORK_DIR: "inference"
SKIP_EXAMPLES: True
DIFFUSION_PARAS:
  SAMPLE:
    VALUES: ['ddim', 'euler', 'euler_ancestral', 'heun', 'dpm2',
             'dpm2_ancestral', 'dpmpp_2m', 'dpmpp_sde', 'dpmpp_2m_sde', 'dpmpp_2s_ancestral',
             'dpm2_karras', 'dpm2_ancestral_karras', 'dpmpp_2s_ancestral_karras', 'dpmpp_2m_karras',
             'dpmpp_sde_karras', 'dpmpp_2m_sde_karras']
    DEFAULT: 'dpmpp_2s_ancestral'
    VISIBLE: True
  NEGATIVE_PROMPT:
    DEFAULT:
    VISIBLE: True
  PROMPT_PREFIX:
    DEFAULT:
    VISIBLE: True
  SAMPLES:
    MIN: 1
    MAX: 4
    DEFAULT: 1
    VISIBLE: True
  NUM_FRAMES:
    MIN: 1
    MAX: 100
    DEFAULT: 49
    VISIBLE: False
  FPS:
    MIN: 1
    MAX: 50
    DEFAULT: 8
    VISIBLE: False
  SAMPLE_STEPS:
    MIN: 1
    MAX: 100
    DEFAULT: 30
    VISIBLE: True
  GUIDE_SCALE:
    MIN: 0
    MAX: 10
    DEFAULT: 5.0
    VISIBLE: True
  GUIDE_RESCALE:
    MIN: 0
    MAX: 1.0
    DEFAULT: 0.5
    VISIBLE: True
  DISCRETIZATION:
    VALUES: ["trailing", "leading", "linspace"]
    DEFAULT: "linspace"
    VISIBLE: True
  REFINE_SAMPLERS:
    VALUES: [ 'ddim', 'euler', 'euler_ancestral', 'heun', 'dpm2',
              'dpm2_ancestral', 'dpmpp_2m', 'dpmpp_sde', 'dpmpp_2m_sde', 'dpmpp_2s_ancestral',
              'dpm2_karras', 'dpm2_ancestral_karras', 'dpmpp_2s_ancestral_karras', 'dpmpp_2m_karras',
              'dpmpp_sde_karras', 'dpmpp_2m_sde_karras' ]
    DEFAULT: 'dpmpp_2s_ancestral'
    VISIBLE: True
  REFINE_SAMPLE_STEPS:
    MIN: 0
    MAX: 100
    DEFAULT: 30
    VISIBLE: True
  REFINE_GUIDE_SCALE:
    MIN: 0
    MAX: 10
    DEFAULT: 5.0
    VISIBLE: True
  REFINE_GUIDE_RESCALE:
    MIN: 0
    MAX: 1.0
    DEFAULT: 0.5
    VISIBLE: True
  REFINE_DISCRETIZATION:
    VALUES: [ "trailing", "leading", "linspace" ]
    DEFAULT: "linspace"
    VISIBLE: True
  AESTHETIC_SCORE:
    MIN: 0.0
    MAX: 10.0
    DEFAULT: 6.0
    VISIBLE: True
  NEGATIVE_AESTHETIC_SCORE:
    MIN: 0.0
    MAX: 10.0
    DEFAULT: 2.5
    VISIBLE: True
  REFINE_STRENGTH:
    MIN: 0
    MAX: 1.0
    DEFAULT: 0.15
  RESOLUTIONS:
    VALUES: [
             [512, 512], [768, 768],
             [704, 1408], [704, 1344], [768, 1344],
             [720, 1280],
             [768, 1280], [832, 1216], [832, 1152],
             [896, 1152], [896, 1088], [960, 1088],
             [960, 1024], [1024, 1024], [1024, 960],
             [1088, 960], [1088, 896], [1152, 896],
             [1152, 832], [1216, 832], [1280, 720],
             [1280, 768],
             [1344, 768], [1344, 704], [1408, 704],
             [1472, 704], [1536, 640], [1600, 640],
             [1664, 576], [1728, 576],
             [2048, 2048], [2048, 1920], [1920, 2048],
             [1536, 2560], [2560, 1536], [2560, 1440],
             [2560, 1440],
             [480, 720], [720, 480]
            ]
    DEFAULT: [1024, 1024]
    VISIBLE: True
EXTENSION_PARAS:
  MANTRA_BOOK: scepter/methods/studio/extensions/mantra_book/mantra_book.yaml
  OFFICIAL_TUNERS: scepter/methods/studio/extensions/tuners/official_tuners.yaml
  OFFICIAL_CONTROLLERS: scepter/methods/studio/extensions/controllers/official_controllers.yaml
  TUNER_MANAGER: scepter/methods/studio/tuner_manager/tuner_manager.yaml
CONTROLABLE_ANNOTATORS:
  -
    NAME: "CannyAnnotator"
    TYPE: Canny
    IS_DEFAULT: True
  -
    NAME: "HedAnnotator"
    PRETRAINED_MODEL: "ms://iic/scepter_scedit@annotator/ckpts/ControlNetHED.pth"
    TYPE: Hed
    IS_DEFAULT: False
  -
    NAME: "OpenposeAnnotator"
    BODY_MODEL_PATH: "ms://iic/scepter_scedit@annotator/ckpts/body_pose_model.pth"
    HAND_MODEL_PATH: "ms://iic/scepter_scedit@annotator/ckpts/hand_pose_model.pth"
    TYPE: Openpose
    IS_DEFAULT: False
  -
    NAME: "MidasDetector"
    PRETRAINED_MODEL: "ms://iic/scepter_scedit@annotator/ckpts/dpt_hybrid-midas-501f0c75.pt"
    TYPE: Midas
    IS_DEFAULT: False
  -
    NAME: "ColorAnnotator"
    TYPE: Color
    IS_DEFAULT: False
  -
    NAME: "InvertAnnotator"
    TYPE: Invert-Preprocess
    IS_DEFAULT: False