import os

IPAdapterModelDir = os.path.join(
    os.path.dirname(os.path.abspath(__file__)), "../../checkpoints", "IP-Adapter"
)


MotionDir = os.path.join(
    os.path.dirname(os.path.abspath(__file__)), "../../checkpoints", "motion"
)


MODEL_CFG = {
    "IPAdapter": {
        "ip_image_encoder": os.path.join(IPAdapterModelDir, "models/image_encoder"),
        "ip_ckpt": os.path.join(IPAdapterModelDir, "ip-adapter_sd15.bin"),
        "ip_scale": 1.0,
        "clip_extra_context_tokens": 4,
        "clip_embeddings_dim": 1024,
        "desp": "",
    },
    "IPAdapterPlus": {
        "ip_image_encoder": os.path.join(IPAdapterModelDir, "image_encoder"),
        "ip_ckpt": os.path.join(IPAdapterModelDir, "ip-adapter-plus_sd15.bin"),
        "ip_scale": 1.0,
        "clip_extra_context_tokens": 16,
        "clip_embeddings_dim": 1024,
        "desp": "",
    },
    "IPAdapterPlus-face": {
        "ip_image_encoder": os.path.join(IPAdapterModelDir, "image_encoder"),
        "ip_ckpt": os.path.join(IPAdapterModelDir, "ip-adapter-plus-face_sd15.bin"),
        "ip_scale": 1.0,
        "clip_extra_context_tokens": 16,
        "clip_embeddings_dim": 1024,
        "desp": "",
    },
    "IPAdapterFaceID": {
        "ip_image_encoder": os.path.join(IPAdapterModelDir, "image_encoder"),
        "ip_ckpt": os.path.join(IPAdapterModelDir, "ip-adapter-faceid_sd15.bin"),
        "ip_scale": 1.0,
        "clip_extra_context_tokens": 4,
        "clip_embeddings_dim": 512,
        "desp": "",
    },
    "musev_referencenet": {
        "ip_image_encoder": os.path.join(IPAdapterModelDir, "image_encoder"),
        "ip_ckpt": os.path.join(
            MotionDir, "musev_referencenet/ip_adapter_image_proj.bin"
        ),
        "ip_scale": 1.0,
        "clip_extra_context_tokens": 4,
        "clip_embeddings_dim": 1024,
        "desp": "",
    },
    "musev_referencenet_pose": {
        "ip_image_encoder": os.path.join(IPAdapterModelDir, "image_encoder"),
        "ip_ckpt": os.path.join(
            MotionDir, "musev_referencenet_pose/ip_adapter_image_proj.bin"
        ),
        "ip_scale": 1.0,
        "clip_extra_context_tokens": 4,
        "clip_embeddings_dim": 1024,
        "desp": "",
    },
}