github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-09-06 11:00:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

convert scripts: fix AutoConfig typo (#1512)

Browse Source
This commit is contained in:
cebtenzzre
2023-10-13 14:16:51 -04:00
committed by GitHub
parent 10f9b49313
commit 22de3c56bd
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gpt4all-backend/scripts/convert_bert_hf_to_gguf.py
View File

@@ -45,7 +45,7 @@ gguf_writer = gguf.GGUFWriter(fname_out, gguf.MODEL_ARCH_NAMES[ARCH])
print("gguf: get model metadata")
config = AutoConfig(dir_model)
config = AutoConfig.from_pretrained(dir_model)
block_count = config.num_hidden_layers
gguf_writer.add_name("BERT")

2
gpt4all-backend/scripts/convert_replit_hf_to_gguf.py → gpt4all-backend/scripts/convert_replit_v1_hf_to_gguf.py
View File

@@ -43,7 +43,7 @@ gguf_writer = gguf.GGUFWriter(fname_out, gguf.MODEL_ARCH_NAMES[ARCH])
print("gguf: get model metadata")
config = AutoConfig(dir_model)
config = AutoConfig.from_pretrained(dir_model)
block_count = config.n_layers
gguf_writer.add_name("Replit")