diff --git a/scripts/vlm/mllama_finetune.py b/scripts/vlm/mllama_finetune.py index 149da26829e8..15cd8078fd32 100644 --- a/scripts/vlm/mllama_finetune.py +++ b/scripts/vlm/mllama_finetune.py @@ -33,6 +33,7 @@ --devices=8 --tp=4 --data_type=mock """ + def main(args): """ Main function for setting up and training the MLLama model.