Spaces:
Runtime error
Runtime error
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",
},
}
|