diff --git a/lit_nlp/examples/prompt_debugging/models.py b/lit_nlp/examples/prompt_debugging/models.py
index ba733fe1..ca9c797e 100644
--- 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