Update litgpt/deploy/serve.py

Co-authored-by: Carlos Mocholí <[email protected]>
Lightning-AI · Apr 25, 2024 · 2d69ce5 · 2d69ce5
1 parent bd5ef3b
commit 2d69ce5
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/litgpt/deploy/serve.py b/litgpt/deploy/serve.py
@@ -32,10 +32,7 @@ def __init__(self,
                  max_new_tokens: int = 50) -> None:
 
         if not _LITSERVE_AVAILABLE:
-            raise ImportError(
-                "Serving LitGPT requires the `litserve` library,"
-                " which can be installed via `pip install litserve`."
-            )
+            raise ImportError(str(_LITSERVE_AVAILABLE))
 
         super().__init__()
         self.checkpoint_dir = checkpoint_dir