{ | |
"task": "text-generation", | |
"framework": "pytorch", | |
"preprocessor": { | |
"type": "text-gen-tokenizer", | |
"padding": false | |
}, | |
"model": { | |
"type": "text-generation", | |
"language": "zh" | |
}, | |
"pipeline": { | |
"type": "text-generation" | |
}, | |
"evaluation": { | |
"dataloader": { | |
"batch_size_per_gpu": 2, | |
"workers_per_gpu": 1, | |
"shuffle": false | |
} | |
} | |
} |