MuseV/configs/model/motion_model.py

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",
 },
}