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
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: