From e1c744821b90307274be4f8d4fddf0c945684a37 Mon Sep 17 00:00:00 2001 From: Brett Kuprel Date: Tue, 5 Jul 2022 09:06:27 -0400 Subject: [PATCH] remove seed from replicate --- README.md | 2 +- replicate_predictor.py | 5 +---- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 5b5db81..52495d5 100644 --- a/README.md +++ b/README.md @@ -53,7 +53,7 @@ display(image) ### Interactive -If the model is being used interactively (e.g. in a notebook) `generate_image_stream` can be used to generate a stream of images as the model is decoding. The detokenizer adds a slight delay for each image. Setting `log2_mid_count` to 3 results in a total of `2 ** 3 = 8` generated images. The only valid values for `log2_mid_count` are 0, 1, 2, 3, 4. This is implemented in the colab. +If the model is being used interactively (e.g. in a notebook) `generate_image_stream` can be used to generate a stream of images as the model is decoding. The detokenizer adds a slight delay for each image. Setting `log2_mid_count` to 3 results in a total of `2 ** 3 = 8` generated images. The only valid values for `log2_mid_count` are 0, 1, 2, 3, and 4. This is implemented in the colab. ```python image_stream = model.generate_image_stream( diff --git a/replicate_predictor.py b/replicate_predictor.py index 9b70e92..198f4b2 100644 --- a/replicate_predictor.py +++ b/replicate_predictor.py @@ -14,10 +14,6 @@ class ReplicatePredictor(BasePredictor): description='Text', default='Dali painting of WALL·E' ), - seed: int = Input( - description='A positive number will generate reproducible results', - default=-1 - ), grid_size: int = Input( description='Size of the image grid', ge=1, @@ -37,6 +33,7 @@ class ReplicatePredictor(BasePredictor): default=4 ), ) -> Iterator[Path]: + seed = -1 image_stream = self.model.generate_image_stream( text, seed,