From 5e8abca54f5566f33660b07497e010b7142c55b2 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Wed, 23 Apr 2025 13:22:17 -0400
Subject: [PATCH] use axolotl cloud image as base and various fixes

---
 .runpod/Dockerfile       |  2 +-
 .runpod/requirements.txt |  5 -----
 .runpod/src/utils.py     | 13 -------------
 .runpod/tests.json       |  6 +-----
 4 files changed, 2 insertions(+), 24 deletions(-)

diff --git a/.runpod/Dockerfile b/.runpod/Dockerfile
index dfd8e1c16..d688d2b31 100644
--- a/.runpod/Dockerfile
+++ b/.runpod/Dockerfile
@@ -1,4 +1,4 @@
-FROM runpod/pytorch:3.10-2.0.0-117
+FROM axolotlai/axolotl-cloud:main-py3.11-cu124-2.6.0
 
 COPY .runpod/requirements.txt /requirements.txt
 RUN --mount=type=cache,target=/root/.cache/pip \
diff --git a/.runpod/requirements.txt b/.runpod/requirements.txt
index af9a6fdcf..a4a6fe21a 100644
--- a/.runpod/requirements.txt
+++ b/.runpod/requirements.txt
@@ -5,11 +5,6 @@
 # git+https://github.com/runpod/runpod-python.git
 # To learn more, see https://pip.pypa.io/en/stable/reference/requirements-file-format/
 runpod~=1.7.0
-huggingface_hub
 typing-extensions
 pydantic
 pydantic-settings
-hf-transfer
-setuptools
-numpy==2.0.0
-axolotl
diff --git a/.runpod/src/utils.py b/.runpod/src/utils.py
index 6e790a212..8245aecf4 100644
--- a/.runpod/src/utils.py
+++ b/.runpod/src/utils.py
@@ -12,19 +12,6 @@ def get_output_dir(run_id):
     return path
 
 
-# def make_valid_config(input_args):
-#     """
-#     Currently limited by all possible axolotl args, no defaults
-#     :param input_args: dict of input args
-#     """
-#     all_args = yaml.safe_load(open("config/config.yaml", "r"))
-#     if not input_args:
-#         print("No args provided, using defaults")
-#     else:
-#         all_args.update(input_args)
-#     return all_args
-
-
 def make_valid_config(input_args):
     """
     Creates and saves updated config file, returns the path to the new config
diff --git a/.runpod/tests.json b/.runpod/tests.json
index c4808cd54..09f796b11 100644
--- a/.runpod/tests.json
+++ b/.runpod/tests.json
@@ -83,11 +83,7 @@
       "12.7",
       "12.6",
       "12.5",
-      "12.4",
-      "12.3",
-      "12.2",
-      "12.1",
-      "12.0"
+      "12.4"
     ]
   }
 }