knox 修订了这个 Gist . 跳至此修订
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 | |
更新
更早