Dockerfile

FROM huggingface/autotrain-advanced:latest

ENV PATH="/app/miniconda/bin:$PATH"

RUN /app/miniconda/bin/pip install -U git+https://github.com/huggingface/transformers.git && \
    /app/miniconda/bin/pip install -U accelerate peft trl bitsandbytes

CMD ["/bin/bash", "-c", "source /app/miniconda/etc/profile.d/conda.sh && conda activate autotrain && autotrain app --host 0.0.0.0 --port 7860 --workers 1"]