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))
|