Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-25 17:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix GGUF metadata reading for falcon

Browse Source
This commit is contained in:
oobabooga 2023-09-11 15:49:15 -07:00
parent 9331ab4798
commit 78811dd89a
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/metadata_gguf.py
View File

@ -50,7 +50,11 @@ value_type_info = {
def get_single(value_type, file): def get_single(value_type, file):
if value_type == GGUFValueType.STRING: if value_type == GGUFValueType.STRING:
value_length = struct.unpack("<Q", file.read(8))[0] value_length = struct.unpack("<Q", file.read(8))[0]
value = file.read(value_length).decode('utf-8') value = file.read(value_length)
try:
value = value.decode('utf-8')
except:
pass
else: else:
type_str = _simple_value_packing.get(value_type) type_str = _simple_value_packing.get(value_type)
bytes_length = value_type_info.get(value_type) bytes_length = value_type_info.get(value_type)