File size: 281 Bytes
96fe658
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
import os

from swift.llm import TrainArguments, sft_main

os.environ['MAX_PIXELS'] = str(16 * 28 * 28)

if __name__ == '__main__':
    sft_main(
        TrainArguments(
            model='Qwen/Qwen2.5-VL-7B-Instruct', dataset='AI-ModelScope/coco#2000', split_dataset_ratio=0.01))