diff --git a/scripts/train.sh b/scripts/train.sh
index 88868ff..f967ed6 100644
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -14,4 +14,4 @@ fi
 
 mkdir -p "$OUTPUT_DIR"
 
-train_gemma --output-dir "$OUTPUT_DIR"
\ No newline at end of file
+train_gemma --model "google/gemma-3-12b-it" --output-dir "$OUTPUT_DIR"
\ No newline at end of file