This workshop is designed for GDSC AI Team & NetAI Lab.
check Usage for setting up the environment
This workshop covers the following topics:
- Pipeline API for NLP tasks (
transformers_pipeline.ipynb
) - Diffuser API for text-to-image generation (
diffuser.ipynb
) - How to fine-tune a bert-base model on a custom dataset (
transformers_finetune.ipynb
) - How to use the LLM(Llama) (
llama.ipynb
)
create a virtual environment via conda
conda create -n huggingface-api python=3.12
conda activate huggingface-api
conda install nvidia/label/cuda-12.1.0::cuda-toolkit
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
pip install huggingface_hub transformers datasets