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

update replicate files

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-05 05:55:10 -04:00
parent 40b16cde0c
commit a083357225
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
replicate_predictor.py
View File

@ -26,23 +26,25 @@ class ReplicatePredictor(BasePredictor):
description='Set the seed to a positive number for reproducible results', description='Set the seed to a positive number for reproducible results',
default=-1 default=-1
), ),
intermediate_image_count: int = Input( log2_intermediate_image_count: int = Input(
description='Set the number of intermediate images to show while running', description='Set the number of intermediate images to show while running',
choices=[1, 2, 4, 8, 16], ge=0,
default=4 le=4,
default=2
), ),
supercondition_factor: int = Input( log2_supercondition_factor: int = Input(
description='Lower results in a wider variety of images but less agreement with the text', description='Lower results in a wider variety of images but less agreement with the text',
choices=[2, 4, 8, 16, 32, 64], ge=1,
default=8 le=6,
default=4
), ),
) -> Iterator[Path]: ) -> Iterator[Path]:
image_stream = self.model.generate_image_stream( image_stream = self.model.generate_image_stream(
text, text,
seed, seed,
grid_size=grid_size, grid_size=grid_size,
log2_mid_count=log2(intermediate_image_count), log2_mid_count=log2_intermediate_image_count,
log2_supercondition_factor=log2(supercondition_factor), log2_supercondition_factor=log2_supercondition_factor,
is_verbose=True is_verbose=True
) )