Update train_dreambooth_lora_sdxl.py

train_dreambooth_lora_sdxl.py

@@ -1352,12 +1352,14 @@ def main(args):
                 prompt=args.instance_prompt,
                 repo_folder=args.output_dir,
                 vae_path=args.pretrained_vae_model_name_or_path,
+                token=args.hub_token
             )
             upload_folder(
                 repo_id=repo_id,
                 folder_path=args.output_dir,
                 commit_message="End of training",
                 ignore_patterns=["step_*", "epoch_*"],
+                token=args.hub_token
             )
 
     accelerator.end_training()