Fix flags for word_language_model in ci script

Author: framoncg

run_python_examples.sh

@@ -154,11 +154,11 @@ function vision_transformer() {
 }
 
 function word_language_model() {
-  uv run main.py --epochs 1 --dry-run $CUDA_FLAG --mps || error "word_language_model failed"
-  uv run generate.py $CUDA_FLAG --mps || error "word_language_model generate failed"
+  uv run main.py --epochs 1 --dry-run $ACCEL_FLAG || error "word_language_model failed"
+  uv run generate.py $ACCEL_FLAG || error "word_language_model generate failed"
   for model in "RNN_TANH" "RNN_RELU" "LSTM" "GRU" "Transformer"; do
-    uv run main.py --model $model --epochs 1 --dry-run $CUDA_FLAG --mps || error "word_language_model failed"
-    uv run generate.py $CUDA_FLAG --mps || error "word_language_model generate failed"
+    uv run main.py --model $model --epochs 1 --dry-run $ACCEL_FLAG || error "word_language_model failed"
+    uv run generate.py $ACCEL_FLAG || error "word_language_model generate failed"
   done
 }