mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-04-05 03:04:56 +08:00
20 lines
374 B
Python
20 lines
374 B
Python
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
sys.path.append(str(Path(__file__).parent.parent))
|
|
|
|
from finetune.models.utils import get_model_cls
|
|
from finetune.schemas import Args
|
|
|
|
|
|
def main():
|
|
args = Args.parse_args()
|
|
trainer_cls = get_model_cls(args.model_name, args.training_type)
|
|
trainer = trainer_cls(args)
|
|
trainer.fit()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|