proto: blob results

This commit is contained in:
Ettore Di Giacinto 2023-12-11 23:29:11 +00:00
parent 9aa2a7ca13
commit ee721efceb
1 changed files with 25 additions and 2 deletions

View File

@ -13,9 +13,10 @@ service Backend {
rpc LoadModel(ModelOptions) returns (Result) {}
rpc PredictStream(PredictOptions) returns (stream Reply) {}
rpc Embedding(PredictOptions) returns (EmbeddingResult) {}
rpc GenerateImage(GenerateImageRequest) returns (Result) {}
rpc GenerateImage(GenerateImageRequest) returns (BlobResult) {}
rpc AudioTranscription(TranscriptRequest) returns (TranscriptResult) {}
rpc TTS(TTSRequest) returns (Result) {}
rpc TTS(TTSRequest) returns (BlobResult) {}
rpc GenerateVideo(GenerateVideoRequest) returns (BlobResult) {}
rpc TokenizeString(PredictOptions) returns (TokenizationResponse) {}
rpc Status(HealthMessage) returns (StatusResponse) {}
}
@ -141,6 +142,12 @@ message Result {
bool success = 2;
}
message BlobResult {
string message = 1;
bool success = 2;
string blob = 3;
}
message EmbeddingResult {
repeated float embeddings = 1;
}
@ -180,6 +187,22 @@ message GenerateImageRequest {
int32 CLIPSkip = 11;
}
message GenerateVideoRequest {
int32 height = 1;
int32 width = 2;
int32 mode = 3;
int32 step = 4;
int32 seed = 5;
string positive_prompt = 6;
string negative_prompt = 7;
string dst = 8;
string src = 9;
// Diffusers
string EnableParameters = 10;
int32 CLIPSkip = 11;
}
message TTSRequest {
string text = 1;
string model = 2;