tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

remove unused imports

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-05 06:31:01 -04:00
parent 8dad676808
commit 9a4d9dc118
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
min_dalle/min_dalle.py
View File

@ -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())

2
replicate_predictor.py
View File

@ -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