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

convert.py: fix metadata format to sync with LLM_KV_NAMES in llama.cpp

Browse source
This commit is contained in:
brian khuu 2024-05-13 12:26:00 +10:00
parent 4bd6227f46
commit 005c1b48f0
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
convert.py
View file

@ -364,14 +364,15 @@ class Metadata:
metadata = Metadata()
# Assigning values to Metadata attributes if they exist in the JSON file
# This is based on LLM_KV_NAMES mapping in llama.cpp
metadata.name = data.get("general.name")
metadata.author = data.get("general.author")
metadata.version = data.get("general.version")
metadata.url = data.get("general.url")
metadata.description = data.get("general.description")
metadata.license = data.get("general.license")
metadata.source_url = data.get("general.source_url")
metadata.source_hf_repo = data.get("general.source_hf_repo")
metadata.source_url = data.get("general.source.url")
metadata.source_hf_repo = data.get("general.source.huggingface.repository")
return metadata