From 539593061c7aa1d4a3c631596489a5c543c318e4 Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Tue, 8 Nov 2022 18:45:06 +0800
Subject: [PATCH] Add more model names

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 README.md    | 1 +
 unixcoder.py | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 51887ec..0034355 100644
--- 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 />
 
diff --git a/unixcoder.py b/unixcoder.py
index f0c74c7..1edbb26 100644
--- 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: