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

Apply suggestions from code review

Browse source 
Co-authored-by: compilade <git@compilade.net>
This commit is contained in:
Brian 2024-07-11 21:10:51 +10:00 • committed by brian khuu
parent dd14b8fdb1
commit 74383ba6d2
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
convert_hf_to_gguf.py
View file

@ -72,7 +72,7 @@ class Model:
raise TypeError(f"{type(self).__name__!r} should not be directly instantiated") raise TypeError(f"{type(self).__name__!r} should not be directly instantiated")
if metadata is None: if metadata is None:
raise TypeError("authorship metadata must be provided") metadata = gguf.Metadata()
self.dir_model = dir_model self.dir_model = dir_model
self.ftype = ftype self.ftype = ftype
@ -101,9 +101,9 @@ class Model:
logger.info(f"choosing --outtype bf16 from first tensor type ({first_tensor.dtype})") logger.info(f"choosing --outtype bf16 from first tensor type ({first_tensor.dtype})")
self.ftype = gguf.LlamaFileType.MOSTLY_BF16 self.ftype = gguf.LlamaFileType.MOSTLY_BF16
# Fallback to model architecture name if metadata name is still missing # Fallback to model directory name if metadata name is still missing
if self.metadata.name is None: if self.metadata.name is None:
self.metadata.name = gguf.MODEL_ARCH_NAMES[self.model_arch] self.metadata.name = dir_model.name
# Generate parameter weight class (useful for leader boards) if not yet determined # Generate parameter weight class (useful for leader boards) if not yet determined
if self.metadata.parameter_class_attribute is None: if self.metadata.parameter_class_attribute is None: