Salt
commited on
Commit
·
a7c6cad
1
Parent(s):
d69c27c
Update server.py
Browse files
server.py
CHANGED
@@ -54,26 +54,10 @@ class SplitArgs(argparse.Action):
|
|
54 |
parser = argparse.ArgumentParser(
|
55 |
prog="TavernAI Extras", description="Web API for transformers models"
|
56 |
)
|
57 |
-
|
58 |
-
"--port", type=int, help="Specify the port on which the application is hosted"
|
59 |
-
)
|
60 |
-
parser.add_argument(
|
61 |
-
"--listen", action="store_true", help="Host the app on the local network"
|
62 |
-
)
|
63 |
-
parser.add_argument(
|
64 |
-
"--share", action="store_true", help="Share the app on CloudFlare tunnel"
|
65 |
-
)
|
66 |
-
parser.add_argument("--cpu", action="store_true", help="Run the models on the CPU")
|
67 |
parser.add_argument("--summarization-model", help="Load a custom summarization model")
|
68 |
parser.add_argument(
|
69 |
"--classification-model", help="Load a custom text classification model"
|
70 |
-
)
|
71 |
-
parser.add_argument("--captioning-model", help="Load a custom captioning model")
|
72 |
-
parser.add_argument(
|
73 |
-
"--keyphrase-model", help="Load a custom keyphrase extraction model"
|
74 |
-
)
|
75 |
-
parser.add_argument("--prompt-model", help="Load a custom prompt generation model")
|
76 |
-
parser.add_argument("--embedding-model", help="Load a custom text embedding model")
|
77 |
|
78 |
parser.add_argument(
|
79 |
"--enable-modules",
|
|
|
54 |
parser = argparse.ArgumentParser(
|
55 |
prog="TavernAI Extras", description="Web API for transformers models"
|
56 |
)
|
57 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
58 |
parser.add_argument("--summarization-model", help="Load a custom summarization model")
|
59 |
parser.add_argument(
|
60 |
"--classification-model", help="Load a custom text classification model"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
61 |
|
62 |
parser.add_argument(
|
63 |
"--enable-modules",
|