Skip to content

Commit

Permalink
feat set kimchima_cli back
Browse files Browse the repository at this point in the history
Signed-off-by: micost <chuan.z@hotmail.com>
Micost committed May 22, 2024
1 parent a459b16 commit 8faa489
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/kimchima/cmds/kimchima_cli.py
Original file line number Diff line number Diff line change
@@ -14,7 +14,7 @@

import argparse

from kimchima.cmds.auto_cli import CommandCrossEncoder
from kimchima.cmds.auto_cli import CommandAutoModel


def main():
@@ -31,7 +31,7 @@ def main():
parser_auto=subparsers.add_parser("auto", help="auto help")
parser_auto.add_argument("model_name_or_path", default="sentence-transformers/all-MiniLM-L6-v2", help="model name or path")
parser_auto.add_argument("text", help="text str or list of text str")
parser_auto.set_defaults(func=CommandCrossEncoder.auto)
parser_auto.set_defaults(func=CommandAutoModel.auto)

args = parser.parse_args()
args.func(args)

0 comments on commit 8faa489

Please sign in to comment.