| # 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 sentencepiece | |
| # Copy all local files into the container | |
| COPY . . | |
| # Set a writable cache directory inside the app folder | |
| # Set writable cache directory inside the app folder | |
| ENV HF_HOME=/app/hf_cache | |
| ENV HF_DATASETS_CACHE=/app/hf_cache | |
| ENV TRANSFORMERS_CACHE=/app/hf_cache | |
| # Create the cache directory and set correct permissions | |
| RUN mkdir -p /app/hf_cache/hub && chmod -R 777 /app/hf_cache | |
| # Set default command (replace with your training script later) | |
| CMD ["python", "train.py"] | |