From 90cdd1b5b8d0248a1a7a02bfe2c1bb7d0dc5099a Mon Sep 17 00:00:00 2001
From: Sermet Pekin <96650846+SermetPekin@users.noreply.github.com>
Date: Mon, 16 Feb 2026 10:58:39 +0300
Subject: [PATCH] Update test.yml

---
 .github/workflows/test.yml | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 033037d..c7adb32 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -24,10 +24,17 @@ jobs:
 
     - name: Set up uv
       uses: astral-sh/setup-uv@v7
-
-    - name: Install dependencies with uv
+    # Linux: Request 'cpu' extra to force the pytorch-cpu index (prevents libcublas crash)
+    - name: Install dependencies (Linux)
+      if: runner.os == 'Linux'
       run: |
-        uv sync --extra cpu 
+        uv sync --extra cpu
+        uv pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
+
+    # macOS: Standard sync 
+    - name: Install dependencies (macOS)
+      if: runner.os == 'macOS'
+      run: uv sync
 
     - name: Add nanochat to PYTHONPATH (Unix)
       run: |