Revision of Jina-13.py

1 file changed, 11 insertions

Jina-13.py(file created)

		@@ -0,0 +1,11 @@
1	+	from docarray import BaseDoc
2	+
3	+
4	+	class PromptDocument(BaseDoc):
5	+	prompt: str
6	+	max_tokens: int
7	+
8	+
9	+	class ModelOutputDocument(BaseDoc):
10	+	token_id: int
11	+	generated_text: str

Newer Older