Update litgpt/deploy/serve.py

Lightning-AI · Apr 22, 2024 · d7a4ac4 · d7a4ac4
1 parent 8170500
commit d7a4ac4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/litgpt/deploy/serve.py b/litgpt/deploy/serve.py
@@ -114,7 +114,7 @@ def run_server(
             generated text but can also lead to more incoherent texts.
         max_new_tokens: The number of generation steps to take.
         devices: How many devices/GPUs to use.
-        accelerator: The type of accelerator to use. For example, "auto", "cuda", "cpu", or "mps".capitalize
+        accelerator: The type of accelerator to use. For example, "auto", "cuda", "cpu", or "mps".
             The "auto" setting (default) chooses a GPU if available, and otherwise uses a CPU.
         port: The network port number on which the model is configured to be served.
     """