zhibinlu
/

vgcn-bert-distilbert-base-uncased

Feature Extraction

Model card Files Files and versions Community

Zhibin Lu commited on Jul 3, 2023

Commit

6fb1109

·

1 Parent(s): 464c02e

correct example code in readme

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -31,8 +31,7 @@ import transformers as tfr
 tokenizer = tfr.AutoTokenizer.from_pretrained("distilbert-base-uncased")
 # Load VGCN-BERT model
-from transformers.models.vgcn_bert.modeling_vgcn_bert import VGCNBertModel
-model = VGCNBertModel.from_pretrained(
     "zhibinlu/vgcn-bert-distilbert-base-uncased", trust_remote_code=True,
     # # if you already have WordGraphs (torch sparse) and their id_maps,
     # # you can directly instantiate VGCN-BERT model with your WGraphs (support multiple graphs)

 tokenizer = tfr.AutoTokenizer.from_pretrained("distilbert-base-uncased")
 # Load VGCN-BERT model
+model = tfr.AutoModel.from_pretrained(
     "zhibinlu/vgcn-bert-distilbert-base-uncased", trust_remote_code=True,
     # # if you already have WordGraphs (torch sparse) and their id_maps,
     # # you can directly instantiate VGCN-BERT model with your WGraphs (support multiple graphs)