From a083357225638cd55e916a02c8738d911dd08557 Mon Sep 17 00:00:00 2001 From: Brett Kuprel Date: Tue, 5 Jul 2022 05:55:10 -0400 Subject: [PATCH] update replicate files --- replicate_predictor.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/replicate_predictor.py b/replicate_predictor.py index eccea66..897be8e 100644 --- a/replicate_predictor.py +++ b/replicate_predictor.py @@ -26,23 +26,25 @@ class ReplicatePredictor(BasePredictor): description='Set the seed to a positive number for reproducible results', default=-1 ), - intermediate_image_count: int = Input( + log2_intermediate_image_count: int = Input( description='Set the number of intermediate images to show while running', - choices=[1, 2, 4, 8, 16], - default=4 + ge=0, + 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', - choices=[2, 4, 8, 16, 32, 64], - default=8 + ge=1, + le=6, + default=4 ), ) -> Iterator[Path]: image_stream = self.model.generate_image_stream( text, seed, grid_size=grid_size, - log2_mid_count=log2(intermediate_image_count), - log2_supercondition_factor=log2(supercondition_factor), + log2_mid_count=log2_intermediate_image_count, + log2_supercondition_factor=log2_supercondition_factor, is_verbose=True )