diff --git a/clip/model.py b/clip/model.py
index 2a09245..9cf262a 100644
--- a/clip/model.py
+++ b/clip/model.py
@@ -1,6 +1,7 @@
 from collections import OrderedDict
 from typing import Tuple, Union
 
+import numpy as np
 import torch
 import torch.nn.functional as F
 from torch import nn