fffiloni commited on
Commit
d0ea9a2
1 Parent(s): f10f70e

Make sure model repo is created private

Browse files
Files changed (1) hide show
  1. train_dreambooth_lora_sdxl.py +1 -1
train_dreambooth_lora_sdxl.py CHANGED
@@ -793,7 +793,7 @@ def main(args):
793
 
794
  if args.push_to_hub:
795
  repo_id = create_repo(
796
- repo_id=args.hub_model_id or Path(args.output_dir).name, exist_ok=True, token=args.hub_token
797
  ).repo_id
798
 
799
  # Load the tokenizers
 
793
 
794
  if args.push_to_hub:
795
  repo_id = create_repo(
796
+ repo_id=args.hub_model_id or Path(args.output_dir).name, exist_ok=True, private=True, token=args.hub_token
797
  ).repo_id
798
 
799
  # Load the tokenizers