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
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 import os
from PIL import Image from PIL import Image
import numpy import numpy
from torch import LongTensor, FloatTensor from torch import LongTensor
import torch import torch
import json import json
import requests import requests
from typing import Callable, Tuple, Iterator from typing import Iterator
torch.set_grad_enabled(False) torch.set_grad_enabled(False)
torch.set_num_threads(os.cpu_count()) 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 from min_dalle import MinDalle
import tempfile import tempfile
from typing import Iterator from typing import Iterator
from math import log2
from cog import BasePredictor, Path, Input from cog import BasePredictor, Path, Input