# Use an official PyTorch image as the base
FROM pytorch/pytorch:latest

# Set working directory
WORKDIR /app

# Install necessary Python libraries
RUN pip install --no-cache-dir transformers datasets accelerate peft torch torchvision torchaudio

# Copy all local files into the container
COPY . .

# Set a writable cache directory inside the app folder
ENV HF_HOME=/app/hf_cache
RUN mkdir -p /app/hf_cache
# Set default command (replace with your training script later)
CMD ["python", "train.py"]