diff --git a/stable_diffusion/txt2image.py b/stable_diffusion/txt2image.py
index 877acae7..41fcd4f4 100644
--- a/stable_diffusion/txt2image.py
+++ b/stable_diffusion/txt2image.py
@@ -5,6 +5,7 @@ import argparse
 import mlx.core as mx
 from PIL import Image
 from tqdm import tqdm
+import numpy as np
 
 from stable_diffusion import StableDiffusion