import { default as Model, ModelForwardAttributes } from '../models/model';
import { default as BasicTrainer } from './BasicTrainer';
import { ITokeniser } from '../tokeniser/type';
import { SFTDatasetBuilder } from './SFTDatasetBuilder';
import { AdamWOptimizer } from './AdamW';
import { AdamWOptimizerConfig } from './types';
export default class SFTTrainer extends BasicTrainer {
    tokenizer: ITokeniser;
    datasetBuilder: SFTDatasetBuilder;
    constructor(model: Model<ModelForwardAttributes>, tokenizer: ITokeniser, optConfig?: Partial<AdamWOptimizerConfig>, optimizer?: AdamWOptimizer);
}
