Test case in sample_code.ipynb
A suitable conda environment named hft can be created and activated with:
conda env create -f environment.yaml
conda activate hft
Alternatively, upgrade your transformers and accelerate package to latest
pip install -U transformers
pip install -U accelerate