diff --git a/min_dalle/min_dalle.py b/min_dalle/min_dalle.py
index 4f146e8..76b2bd1 100644
--- a/min_dalle/min_dalle.py
+++ b/min_dalle/min_dalle.py
@@ -1,11 +1,11 @@
 import os
 from PIL import Image
 import numpy
-from torch import LongTensor, FloatTensor
+from torch import LongTensor
 import torch
 import json
 import requests
-from typing import Callable, Tuple, Iterator
+from typing import Iterator
 torch.set_grad_enabled(False)
 torch.set_num_threads(os.cpu_count())
 
diff --git a/replicate_predictor.py b/replicate_predictor.py
index 685d99a..85b28a6 100644
--- a/replicate_predictor.py
+++ b/replicate_predictor.py
@@ -1,8 +1,6 @@
-from contextlib import suppress
 from min_dalle import MinDalle
 import tempfile
 from typing import Iterator
-from math import log2
 from cog import BasePredictor, Path, Input