use axolotl cloud image as base and various fixes

.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 \

.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

.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

.runpod/tests.json

@@ -83,11 +83,7 @@
       "12.7",
       "12.6",
       "12.5",
-      "12.4",
+      "12.4"
-      "12.3",
-      "12.2",
-      "12.1",
-      "12.0"
     ]
   }
 }