Lutech-AI
/

I-SPIn

Lutech-admin commited on Jan 12, 2023

Commit

71b8671

1 Parent(s): f419245

Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -66,7 +66,7 @@ model = ISPIn.from_pretrained('[your/path]/I-SPIn')
 ## Full model architecture
 ```markdown
 ISPIn(
-  (encoder): XLMRobertaModel(...) # same architecture as paraphrase-multilingual-mpnet-base-v2
   (layers): ModuleList(
     (0): Linear(in_features=1536, out_features=1024, bias=True)
     (1): Linear(in_features=1024, out_features=512, bias=True)

 ## Full model architecture
 ```markdown
 ISPIn(
+  (encoder): XLMRobertaModel(...) # transformers internal implementation of paraphrase-multilingual-mpnet-base-v2
   (layers): ModuleList(
     (0): Linear(in_features=1536, out_features=1024, bias=True)
     (1): Linear(in_features=1024, out_features=512, bias=True)