Include revision

huggingface · Apr 17, 2024 · 30153d5 · 30153d5
1 parent 145f1fe
commit 30153d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/text-generation-inference/server/text_generation_server/cli.py b/text-generation-inference/server/text_generation_server/cli.py
@@ -59,7 +59,7 @@ def serve(
     from .server import serve
 
     model_path = fetch_model(model_id, revision)
-    serve(model_path, uds_path)
+    serve(model_path, revision, uds_path)
 
 
 @app.command()