convert scripts: fix AutoConfig typo (#1512)

mmcleanup
cebtenzzre 12 months ago committed by GitHub
parent 10f9b49313
commit 22de3c56bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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")

@ -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")
Loading…
Cancel
Save