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

update replicate

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-11 13:19:06 -04:00
parent 355ef19157
commit f875572a9b
2 changed files with 26 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
min_dalle.ipynb vendored
View File

File diff suppressed because one or more lines are too long

22
replicate_predictor.py
View File

@ -27,22 +27,22 @@ class ReplicatePredictor(BasePredictor):
le=8,
default=4
),
log2_supercondition_factor: int = Input(
description='Higher values result in better agreement with the text but a narrower variety of generated images',
ge=1,
le=6,
default=4
log2_temperature: float = Input(
description='A higher temperature results in more variety.',
ge=-3,
le=3,
default=0
),
) -> Iterator[Path]:
try:
seed = -1
log2_mid_count = 3 if intermediate_outputs else 0
image_stream = self.model.generate_image_stream(
text,
seed,
text = text,
seed = -1,
grid_size = grid_size,
log2_mid_count=log2_mid_count,
log2_supercondition_factor=log2_supercondition_factor,
log2_mid_count = 3 if intermediate_outputs else 0,
temperature = 2 ** log2_temperature,
supercondition_factor = 2 ** 4,
top_k = 2 ** 8,
is_verbose = True
)