diff --git a/gpt4all-backend/scripts/convert_bert_hf_to_gguf.py b/gpt4all-backend/scripts/convert_bert_hf_to_gguf.py index 16d6f457..ea0560d9 100755 --- a/gpt4all-backend/scripts/convert_bert_hf_to_gguf.py +++ b/gpt4all-backend/scripts/convert_bert_hf_to_gguf.py @@ -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") diff --git a/gpt4all-backend/scripts/convert_replit_hf_to_gguf.py b/gpt4all-backend/scripts/convert_replit_v1_hf_to_gguf.py similarity index 98% rename from gpt4all-backend/scripts/convert_replit_hf_to_gguf.py rename to gpt4all-backend/scripts/convert_replit_v1_hf_to_gguf.py index 125df6dd..43cb4871 100755 --- a/gpt4all-backend/scripts/convert_replit_hf_to_gguf.py +++ b/gpt4all-backend/scripts/convert_replit_v1_hf_to_gguf.py @@ -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")