Starting the UI

The AutoTrain UI can be started in multiple ways depending on your needs. We offer UI on Hugging Face Spaces, Colab and locally!

Hugging Face Spaces

To start the UI on Hugging Face Spaces, you can simply click on the following link:

Please make sure you keep the space private and attach appropriate hardware to the space. You can also read more about AutoTrain on the homepage and follow the link there to start your own training instance on Hugging Face Spaces. Click here to visit the homepage.

Colab

To start the UI on Colab, you can simply click on the following link:

Please note, to run the app on Colab, you will need an ngrok token. You can get one by signing up for free on ngrok. This is because Colab does not allow exposing ports to the internet directly.

Locally

To run the autotrain app locally, install autotrain-advanced python package:

$ pip install autotrain-advanced

and then run the following command:

$ export HF_TOKEN=your_hugging_face_write_token
$ autotrain app --host 127.0.0.1 --port 8000

This will start the app on http://127.0.0.1:8000.

AutoTrain doesn’t install pytorch, torchaudio, torchvision, or any other dependencies. You will need to install them separately. It is thus recommended to use conda environment:

$ conda create -n autotrain python=3.10
$ conda activate autotrain

$ pip install autotrain-advanced

$ conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
$ conda install -c "nvidia/label/cuda-12.1.0" cuda-nvcc
$ conda install xformers -c xformers

$ python -m nltk.downloader punkt
$ pip install flash-attn --no-build-isolation
$ pip install deepspeed

$ export HF_TOKEN=your_hugging_face_write_token
$ autotrain app --host 127.0.0.1 --port 8000

In case of any issues, please report on the GitHub issues.