vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix linting

Browse source
This commit is contained in:
Christian Zhou-Zheng 2024-06-09 11:23:55 -04:00
parent 0779f2f74f
commit a234bf821b
2 changed files with 8 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

10
convert-hf-to-gguf.py
View file

@ -2891,15 +2891,13 @@ def main() -> None:
model_instance.gguf_writer.add_quantization_version(gguf.GGML_QUANT_VERSION)
if args.vocab_only:
logger.info(f"Exporting model vocab...")
logger.info("Exporting model vocab...")
model_instance.write_vocab()
logger.info(f"Model vocab successfully exported.")
logger.info("Model vocab successfully exported.")
else:
logger.info(f"Exporting model...")
logger.info("Exporting model...")
model_instance.write()
logger.info(f"Model successfully exported.")
logger.info("Model successfully exported.")
if __name__ == '__main__':
main()

1
gguf-py/gguf/gguf_writer.py
View file

@ -30,7 +30,6 @@ from .constants import (
)
from .quants import quant_shape_from_byte_shape
from .constants import Keys
logger = logging.getLogger(__name__)