diff --git a/src/axolotl/utils/trainer.py b/src/axolotl/utils/trainer.py
index ddc22b454..8ba12cbbf 100644
--- a/src/axolotl/utils/trainer.py
+++ b/src/axolotl/utils/trainer.py
@@ -12,6 +12,7 @@ from typing import Optional, Union
 
 import numpy as np
 import torch.cuda
+import transformers
 from datasets import Dataset, set_caching_enabled
 from torch.optim.lr_scheduler import OneCycleLR
 from torch.utils.data import (