This repository contains a Dockerfile for AutoTrain Advanced to run with Kubernetes pods. Also along with AutoTrain comes JupyterLab.

• GitHub Repo
• AutoTrain Repo
• Find HF_TOKEN

----------------------------------------------
HF_TOKEN         = (YOUR HUGGINGFACE TOKEN) !!
JUPYTER_PASSWORD = (PASSWORD)               !
----------------------------------------------
AUTOTRAIN_PORT   = 7860 (default)
JUPYTER_PORT     = 8888 (default)
----------------------------------------------

If HF_TOKEN is not set, AutoTrain will not start properly. Same with Jupyter.

• 7860/tcp (AutoTrain UI)
• 8888/tcp (JupyterLab)

buildx bake

cd container

docker buildx bake

docker buildx bake --push

docker build should be ran from the root of the repository, not from the container folder. The build command should be ran as follows:

docker build -t r4mnd/container:<version> -f container/Dockerfile .