Spaces:
Running
Running
Update train_dreambooth_b-lora_sdxl.py
Browse files
train_dreambooth_b-lora_sdxl.py
CHANGED
@@ -1027,6 +1027,7 @@ def main(args):
|
|
1027 |
if args.push_to_hub:
|
1028 |
repo_id = create_repo(
|
1029 |
repo_id=args.hub_model_id or Path(args.output_dir).name,
|
|
|
1030 |
exist_ok=True,
|
1031 |
token=args.hub_token,
|
1032 |
).repo_id
|
|
|
1027 |
if args.push_to_hub:
|
1028 |
repo_id = create_repo(
|
1029 |
repo_id=args.hub_model_id or Path(args.output_dir).name,
|
1030 |
+
private=True,
|
1031 |
exist_ok=True,
|
1032 |
token=args.hub_token,
|
1033 |
).repo_id
|