]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
server : add support for the RPC backend (#7305)
authorRadoslav Gerganov <redacted>
Fri, 17 May 2024 07:00:17 +0000 (10:00 +0300)
committerGitHub <redacted>
Fri, 17 May 2024 07:00:17 +0000 (10:00 +0300)
ref: #7292

examples/server/server.cpp

index 7e0d068f830abb2659d7696f8d89e663eec85365..b598076d315bd18d725d6c7828d9a7397a7dea39 100644 (file)
@@ -2387,6 +2387,7 @@ static void server_print_usage(const char * argv0, const gpt_params & params, co
     printf("  --lora-base FNAME         optional model to use as a base for the layers modified by the LoRA adapter\n");
     printf("  --host                    ip address to listen (default  (default: %s)\n", sparams.hostname.c_str());
     printf("  --port PORT               port to listen (default  (default: %d)\n", sparams.port);
+    printf("  --rpc SERVERS             comma separated list of RPC servers\n");
     printf("  --path PUBLIC_PATH        path from which to serve static files (default: disabled)\n");
     printf("  --api-key API_KEY         optional api key to enhance server security. If set, requests must include this key for access.\n");
     printf("  --api-key-file FNAME      path to file containing api keys delimited by new lines. If set, requests must include one of the keys for access.\n");
@@ -2439,6 +2440,12 @@ static void server_params_parse(int argc, char ** argv, server_params & sparams,
                 break;
             }
             sparams.port = std::stoi(argv[i]);
+        } else if (arg == "--rpc") {
+            if (++i >= argc) {
+                invalid_param = true;
+                break;
+            }
+            params.rpc_servers = argv[i];
         } else if (arg == "--host") {
             if (++i >= argc) {
                 invalid_param = true;