Archives/gpt4all
2
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all synced 2024-11-16 06:13:09 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats 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") print("gguf: get model metadata")
config = AutoConfig(dir_model) config = AutoConfig.from_pretrained(dir_model)
block_count = config.num_hidden_layers block_count = config.num_hidden_layers
gguf_writer.add_name("BERT") 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") print("gguf: get model metadata")
config = AutoConfig(dir_model) config = AutoConfig.from_pretrained(dir_model)
block_count = config.n_layers block_count = config.n_layers
gguf_writer.add_name("Replit") gguf_writer.add_name("Replit")