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