File size: 638 Bytes
96d7ad8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import os


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


MODEL_CFG = {
    "musev": {
        "unet": os.path.join(MotionDIr, "musev"),
        "desp": "only train unet motion module, fix t2i",
    },
    "musev_referencenet": {
        "unet": os.path.join(MotionDIr, "musev_referencenet"),
        "desp": "train referencenet, IPAdapter and unet motion module, fix t2i",
    },
    "musev_referencenet_pose": {
        "unet": os.path.join(MotionDIr, "musev_referencenet_pose"),
        "desp": "train  unet motion module and IPAdapter, fix t2i and referencenet",
    },
}