mirror of
https://github.com/THUDM/CogVideo.git
synced 2025-04-06 03:57:56 +08:00
60f6a3d7ee
- Add Trainer base class with core training loop functionality - Implement distributed training setup with Accelerate - Add training script with model/trainer initialization - Support LoRA fine-tuning with checkpointing and validation
19 lines
373 B
Python
19 lines
373 B
Python
import sys
|
|
from pathlib import Path
|
|
|
|
sys.path.append(str(Path(__file__).parent.parent))
|
|
|
|
from finetune.schemas import Args
|
|
from finetune.models.utils import get_model_cls
|
|
|
|
|
|
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()
|