diff --git a/mfai/torch/models/encoders.py b/mfai/torch/models/encoders.py
index 7e7f821..de2ea7b 100644
--- a/mfai/torch/models/encoders.py
+++ b/mfai/torch/models/encoders.py
@@ -4,7 +4,6 @@
 import torch
 import torch.nn as nn
 import torch.utils.model_zoo as model_zoo
-import torchvision.models as models
 from dataclasses_json import dataclass_json
 from torchvision.models.resnet import BasicBlock, Bottleneck, ResNet