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) model_instance.gguf_writer.add_quantization_version(gguf.GGML_QUANT_VERSION)
if args.vocab_only: if args.vocab_only:
logger.info(f"Exporting model vocab...") logger.info("Exporting model vocab...")
model_instance.write_vocab() model_instance.write_vocab()
logger.info(f"Model vocab successfully exported.") logger.info("Model vocab successfully exported.")
else: else:
logger.info(f"Exporting model...") logger.info("Exporting model...")
model_instance.write() model_instance.write()
logger.info(f"Model successfully exported.") logger.info("Model successfully exported.")
if __name__ == '__main__': if __name__ == '__main__':
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 .quants import quant_shape_from_byte_shape
from .constants import Keys
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)