Update convert_hf_to_gguf.py

Co-authored-by: compilade <git@compilade.net>
This commit is contained in:
Robert Sinclair 2024-07-28 02:03:28 +03:00 committed by GitHub
parent 7ff30834e1
commit 9758c5a92b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -3678,7 +3678,8 @@ def main() -> None:
metadata_override=args.metadata, model_name=args.model_name, metadata_override=args.metadata, model_name=args.model_name,
split_max_tensors=args.split_max_tensors, split_max_tensors=args.split_max_tensors,
split_max_size=split_str_to_n_bytes(args.split_max_size), dry_run=args.dry_run, split_max_size=split_str_to_n_bytes(args.split_max_size), dry_run=args.dry_run,
small_first_shard=args.no_tensor_first_split,z=args.z) small_first_shard=args.no_tensor_first_split,
z=args.z)
if args.vocab_only: if args.vocab_only:
logger.info("Exporting model vocab...") logger.info("Exporting model vocab...")