properly limit input to 64 tokens

This commit is contained in:
Brett Kuprel
2022-07-05 22:14:19 -04:00
parent 1a8c01047c
commit f071b31bdd
4 changed files with 6 additions and 4 deletions
+1 -1
View File
@@ -11,7 +11,7 @@ class ReplicatePredictor(BasePredictor):
def predict(
self,
text: str = Input(
description='Text',
description='For long prompts, only the first 64 tokens will be used to generate the image.',
default='Dali painting of WALL·E'
),
intermediate_outputs: bool = Input(