jonghyunlee
/

ChemBERT_ChEMBL_pretrained

Feature Extraction

text-embeddings-inference

Model card Files Files and versions

jonghyunlee commited on Sep 6, 2023

Commit

7e55a0d

·

1 Parent(s): 3c47826

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ The model architecture utilized is based on BERT. Here are the key configuration
 ```
 BertConfig(
-    vocab_size=len(tokenizer_pretrained.vocab),
     hidden_size=256,
     num_hidden_layers=8,
     num_attention_heads=8,

 ```
 BertConfig(
+    vocab_size=70,
     hidden_size=256,
     num_hidden_layers=8,
     num_attention_heads=8,