File size: 1,155 Bytes
14ac044 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | # 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"]
|