Update convert-hf-to-gguf.py

Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
This commit is contained in:
slaren 2024-03-31 19:52:46 +02:00 committed by GitHub
parent 3b3298af17
commit 8c2f7b8169
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1268,7 +1268,7 @@ class LlamaModel(Model):
if self.ftype == 1 and data_dtype == np.float32:
data = data.astype(np.float16)
merged_name = f"layers.{bid}.feed_forward.experts.w{wid}"
merged_name = f"layers.{bid}.feed_forward.experts.w{wid}.weight"
new_name = tensor_map.get_name(merged_name, try_suffixes=(".weight", ".bias"))
if new_name is None: