n_layers_per_step = 1, last_layer_weight = 1.5 * model_layers,, prior_layers_weight= 1, kl_div_weight = 2, kl_temperature= 1, 554d487 verified bobox commited on Jun 12, 2024