sonoisa
/

sentence-t5-base-ja-mean-tokens

Feature Extraction

sentence-transformers

sentence-similarity

Inference Endpoints

Model card Files Files and versions Community

sonoisa commited on Jul 28, 2022

Commit

72d9b3d

•

1 Parent(s): 39f060c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ import torch
 class SentenceT5:
     def __init__(self, model_name_or_path, device=None):
-        self.tokenizer = T5Tokenizer.from_pretrained(model_name_or_path)
         self.model = T5Model.from_pretrained(model_name_or_path).encoder
         self.model.eval()

 class SentenceT5:
     def __init__(self, model_name_or_path, device=None):
+        self.tokenizer = T5Tokenizer.from_pretrained(model_name_or_path, is_fast=False)
         self.model = T5Model.from_pretrained(model_name_or_path).encoder
         self.model.eval()