update readme
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user