This video is a hands-on step-by-step tutorial to install AutoTrain by hugging face on local system and then fine-tune any model on custom dataset.
$ 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
$ export HF_TOKEN=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
$ autotrain app --host 0.0.0.0 --port 8000
Dataset Used:
text,label
"This movie was fantastic!",positive
"The food at that restaurant was terrible.",negative
"I love this song!",positive
"The customer service was excellent.",positive
"I couldn't finish reading this book.",negative
No comments:
Post a Comment