Use Tailscale endpoints, add RunPod Docker build files

components/runpod_trainer/Dockerfile

@@ -2,20 +2,17 @@ FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0-devel-ubuntu22.04
 
 WORKDIR /app
 
+# Copy requirements first for better caching
+COPY requirements.txt /app/requirements.txt
+
 # Install dependencies
-RUN pip install --no-cache-dir \
-    runpod \
-    transformers \
-    datasets \
-    accelerate \
-    boto3 \
-    scikit-learn \
-    scipy
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy handler
 COPY handler.py /app/handler.py
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
+ENV HF_HOME=/tmp/huggingface
 
 CMD ["python", "-u", "handler.py"]

components/runpod_trainer/requirements.txt

@@ -0,0 +1,9 @@
+runpod>=1.6.0
+transformers>=4.36.0
+datasets>=2.16.0
+accelerate>=0.25.0
+boto3>=1.34.0
+scikit-learn>=1.3.0
+scipy>=1.11.0
+torch>=2.1.0
+safetensors>=0.4.0