update readme

This commit is contained in:
Brett Kuprel
2022-07-17 15:29:23 -04:00
parent f0c4fc7350
commit cdcf3d3964
2 changed files with 13 additions and 12 deletions
+7 -9
View File
@@ -1,3 +1,4 @@
from min_dalle import MinDalle
import tempfile
import torch, torch.backends.cudnn
@@ -22,13 +23,10 @@ class ReplicatePredictor(BasePredictor):
progressive_outputs: bool = Input(default=True),
seamless: bool = Input(default=False),
grid_size: int = Input(ge=1, le=9, default=5),
temperature: str = Input(
choices=(
['1/{}'.format(2 ** i) for i in range(4, 0, -1)] +
[str(2 ** i) for i in range(5)]
),
default='4',
description='Advanced Setting, see Readme below if interested.'
temperature: float = Input(
ge=0.01,
le=16,
default=4
),
top_k: int = Input(
choices=[2 ** i for i in range(15)],
@@ -46,8 +44,8 @@ class ReplicatePredictor(BasePredictor):
seed = -1,
grid_size = grid_size,
progressive_outputs = progressive_outputs,
is_seamless=seamless,
temperature = eval(temperature),
is_seamless = seamless,
temperature = temperature,
supercondition_factor = float(supercondition_factor),
top_k = top_k,
is_verbose = True