Add more model names

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 539593061c
2 changed files with 3 additions and 1 deletions
--- a/README.md
+++ b/README.md
@ -66,6 +66,7 @@ The default value is None, which enables GPU if cuda is available.

 Supported model names:
 - microsoft/unixcoder-base
+- microsoft/unixcoder-base-nine

 <br />

--- a/unixcoder.py
+++ b/unixcoder.py
@ -149,7 +149,8 @@ class Unixcoder(NNOperator):
    @staticmethod
    def supported_model_names(format: str = None):
        full_list = [
-            'microsoft/unixcoder-base'
+            'microsoft/unixcoder-base',
+            'microsoft/unixcoder-base-nine'
        ]
        full_list.sort()
        if format is None: