def gen_pre_train():
    gens.train(gen_config)
def gen_pre_train(text_data):
    gens.train(gen_config, text_data)