adjust fine_tuning to read dataset from model