diff --git a/s_bert.py b/s_bert.py index 429144a..f87130e 100644 --- a/s_bert.py +++ b/s_bert.py @@ -199,6 +199,7 @@ class STransformers(NNOperator): 'sentence-t5-xxl', 'sentence-t5-xl', 'sentence-t5-large', + 'sentence-t5-base', 'all-distilroberta-v1', 'gtr-t5-xxl', 'gtr-t5-large',