sayakpaul HF Staff commited on
Commit
8d702cb
·
1 Parent(s): 9959fea
Files changed (1) hide show
  1. convert_diffusion_to_gguf.py +6 -1
convert_diffusion_to_gguf.py CHANGED
@@ -308,7 +308,12 @@ def convert(args):
308
 
309
  logging.info("Hub repo ID detected.")
310
  allow_patterns = f"{args.subfolder}/*.*" if args.subfolder else None
311
- local_dir = snapshot_download(repo_id=str(args.model), local_dir=args.cache_dir, allow_patterns=allow_patterns)
 
 
 
 
 
312
  repo_id = str(args.model)
313
  local_dir = Path(local_dir)
314
  local_dir = local_dir / args.subfolder if args.subfolder else local_dir
 
308
 
309
  logging.info("Hub repo ID detected.")
310
  allow_patterns = f"{args.subfolder}/*.*" if args.subfolder else None
311
+ local_dir = snapshot_download(
312
+ repo_id=str(args.model),
313
+ local_dir=args.cache_dir,
314
+ allow_patterns=allow_patterns,
315
+ token=args.hf_token
316
+ )
317
  repo_id = str(args.model)
318
  local_dir = Path(local_dir)
319
  local_dir = local_dir / args.subfolder if args.subfolder else local_dir