Fix a bug in the prompt debugging demo.

PiperOrigin-RevId: 645466395
PAIR-code · Jun 21, 2024 · e0e35c3 · e0e35c3
1 parent afd51fe
commit e0e35c3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lit_nlp/examples/prompt_debugging/models.py b/lit_nlp/examples/prompt_debugging/models.py
@@ -12,7 +12,7 @@
 DEFAULT_BATCH_SIZE = 1
 DEFAULT_DL_FRAMEWORK = "kerasnlp"
 DEFAULT_DL_RUNTIME = "tensorflow"
-DEFAULT_MODELS = ["gemma_1.1_instruct_2b_en:gemma_1.1_instruct_2b_en"],
+DEFAULT_MODELS = ["gemma_1.1_instruct_2b_en:gemma_1.1_instruct_2b_en"]
 DEFAULT_PRECISION = "bfloat16"
 
 

diff --git a/lit_nlp/examples/prompt_debugging/server.py b/lit_nlp/examples/prompt_debugging/server.py
@@ -8,7 +8,7 @@
 To run with the default configuration (Gemma on TensorFlow via Keras):
 
   python3 -m lit_nlp.examples.prompt_debugging.server -- \
-    --models=gemma:gemma_1.1_instruct_2b_en \
+    --models=gemma:gemma_1.1_instruct_2b_en, \
     --alsologtostderr
 
 MODELS: