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