Spaces:
No application file
No application file
File size: 2,216 Bytes
6755a2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
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": "",
},
}
|