(logger_name,output_dir)

(base_model, load_8bit=False)

()

( input_ids: torch.Tensor, model: torch.nn.Module, tokenizer: transformers.PreTrainedTokenizer,

(prompt)

(data_point)

( train_on_inputs: bool = False, # if False, masks out inputs in loss group_by_length: bool = True,