Spaces:
Running
Running
Update train_dreambooth_b-lora_sdxl.py
Browse files
train_dreambooth_b-lora_sdxl.py
CHANGED
@@ -2019,6 +2019,7 @@ def main(args):
|
|
2019 |
folder_path=args.output_dir,
|
2020 |
commit_message="End of training",
|
2021 |
ignore_patterns=["step_*", "epoch_*"],
|
|
|
2022 |
)
|
2023 |
|
2024 |
accelerator.end_training()
|
|
|
2019 |
folder_path=args.output_dir,
|
2020 |
commit_message="End of training",
|
2021 |
ignore_patterns=["step_*", "epoch_*"],
|
2022 |
+
token=args.hub_token
|
2023 |
)
|
2024 |
|
2025 |
accelerator.end_training()
|