From 9a4d9dc1182afc9fa48be2afc8a02e3c1bb2667f Mon Sep 17 00:00:00 2001
From: Brett Kuprel <brkuprel@gmail.com>
Date: Tue, 5 Jul 2022 06:31:01 -0400
Subject: [PATCH] remove unused imports

---
 min_dalle/min_dalle.py | 4 ++--
 replicate_predictor.py | 2 --
 2 files changed, 2 insertions(+), 4 deletions(-)

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