| # Hayat Elixir AI V5.1 - Meditron3-70B QLoRA Training | |
| # ERROR-PROOF v2 — addresses all 17 known HF Space errors | |
| # ERR-17 FIX: Disk-safe model loading with shard cleanup | |
| FROM nvidia/cuda:12.1.0-devel-ubuntu22.04 | |
| ENV DEBIAN_FRONTEND=noninteractive | |
| ENV HF_HOME=/app/hf_cache | |
| ENV TRANSFORMERS_CACHE=/app/hf_cache | |
| ENV TORCH_HOME=/app/torch_cache | |
| ENV PYTHONUNBUFFERED=1 | |
| # Faster downloads from HuggingFace Hub | |
| ENV HF_HUB_ENABLE_HF_TRANSFER=1 | |
| RUN apt-get update && apt-get install -y \ | |
| python3 python3-pip git wget curl \ | |
| && rm -rf /var/lib/apt/lists/* | |
| WORKDIR /app | |
| RUN chmod -R 777 /app | |
| # ERR-07: PyTorch >= 2.4 | |
| RUN pip3 install --no-cache-dir \ | |
| torch==2.5.1 --index-url https://download.pytorch.org/whl/cu121 | |
| # ERR-09/10: Pin ALL versions, include rich | |
| RUN pip3 install --no-cache-dir \ | |
| transformers==4.46.0 \ | |
| peft==0.13.0 \ | |
| trl==0.9.6 \ | |
| bitsandbytes==0.44.1 \ | |
| accelerate==1.0.0 \ | |
| datasets==3.0.0 \ | |
| huggingface_hub==0.26.0 \ | |
| hf_transfer \ | |
| rich flask scipy sentencepiece protobuf | |
| COPY train.py /app/ | |
| COPY hayat_writer_v5_MASTER.jsonl /app/ | |
| RUN chmod -R 777 /app | |
| EXPOSE 7860 | |
| CMD ["python3", "train.py"] | |