From 1219a239d2a5d43221ffe8b065474299f16b5667 Mon Sep 17 00:00:00 2001
From: Gregory Kielian <gkielian@google.com>
Date: Mon, 6 Nov 2023 16:52:25 -0800
Subject: [PATCH] test commit 3

---
 .github/workflows/shakespeare-char-test.yml | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/.github/workflows/shakespeare-char-test.yml b/.github/workflows/shakespeare-char-test.yml
index 769842cf5..708524e72 100644
--- a/.github/workflows/shakespeare-char-test.yml
+++ b/.github/workflows/shakespeare-char-test.yml
@@ -15,5 +15,13 @@ jobs:
       - name: List files in the repository
         run: |
           ls ${{ github.workspace }}
+      - name: install deps for cpu inference
+        run: |
+          python3 -m pip install --upgrade pip
+          pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+          pip3 install torch numpy transformers datasets tiktoken wandb tensorboard tqdm
+      - name: run small network for cpu inference test
+        run: |
+          python3 train.py --device=cpu --eval_iters=1 --log_interval=1 --block_size=2 --batch_size=2 --n_layer=2 --n_head=2 --n_embd=16 --max_iters=2 --lr_decay_iters=2 --dropout=0.0
       - run: echo "🍏 This job's status is ${{ job.status }}."