training_args = TrainingArguments(output_dir="my-awesome-model", push_to_hub=True)

Pass your training arguments as usual to [Trainer]:

trainer = Trainer(
     model=model,
     args=training_args,
     train_dataset=small_train_dataset,
     eval_dataset=small_eval_dataset,
     compute_metrics=compute_metrics,
 )