LIT:Update default DL runtime to torch.

PiperOrigin-RevId: 681896534
PAIR-code · Oct 3, 2024 · e2513e8 · e2513e8
1 parent bc641d7
commit e2513e8
Showing 1 changed file with 1 addition and 1 deletion.
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_DL_RUNTIME = "torch"
 DEFAULT_MODELS = ["gemma_1.1_instruct_2b_en:gemma_1.1_instruct_2b_en"]
 DEFAULT_PRECISION = "bfloat16"
 DEFAULT_SEQUENCE_LENGTH = 512