From e0011fdf55b5c1276e934143313fa7980435025b Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Wed, 31 May 2023 09:52:03 -0400
Subject: [PATCH] Fix base builder, missing tags

---
 .github/workflows/base.yml | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/base.yml b/.github/workflows/base.yml
index b6ab3a04f..c5a70978b 100644
--- a/.github/workflows/base.yml
+++ b/.github/workflows/base.yml
@@ -17,10 +17,12 @@ jobs:
           - cuda: "118"
             cuda_version: 11.8.0
             python_version: "3.9"
+            pytorch: 2.0.0
             axolotl_extras:
           - cuda: "118"
             cuda_version: 11.8.0
             python_version: "3.10"
+            pytorch: 2.0.0
             axolotl_extras:
           - cuda: "117"
             cuda_version: 11.7.0
@@ -28,8 +30,8 @@ jobs:
             pytorch: 1.13.1
             axolotl_extras:
           - cuda: "118"
-            python_version: "3.9"
             cuda_version: 11.8.0
+            python_version: "3.9"
             pytorch: 2.0.0
             axolotl_extras: gptq
     steps: