| # 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 default command (replace with your training script later) | |
| CMD ["python", "train.py"] | |