faster decoder self attention

This commit is contained in:
Brett Kuprel
2022-07-04 08:05:55 -04:00
parent a79d30f718
commit 377d15cb16
4 changed files with 21 additions and 23 deletions

View File

@@ -13,16 +13,16 @@ class Predictor(BasePredictor):
description='Text',
default='Dali painting of WALL·E'
),
seed: int = Input(
description='Set the seed to a positive number for reproducible results',
default=-1
),
grid_size: int = Input(
description='Size of the image grid',
ge=1,
le=4,
default=4
)
),
seed: int = Input(
description='Set the seed to a positive number for reproducible results',
default=-1
),
) -> Path:
image = self.model.generate_image(text, seed, grid_size=grid_size)
out_path = Path(tempfile.mkdtemp()) / 'output.jpg'