strips/stable-diffusion-rpc
2
0
Fork 0
mirror of https://github.com/GayPizzaSpecifications/stable-diffusion-rpc.git synced 2025-08-04 14:01:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Start work on C++ client, and implement streaming of image generation.

Browse Source
This commit is contained in:
Alex Zenla
2023-04-23 14:22:10 -07:00
parent 1bb629c18f
commit b063d91b1e
11 changed files with 509 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
Common/StableDiffusion.proto
View File

@ -270,6 +270,29 @@ message GenerateImagesResponse {
repeated uint32 seeds = 2;
}
message GenerateImagesBatchProgressUpdate {
float percentage_complete = 1;
}
message GenerateImagesBatchCompletedUpdate {
repeated Image images = 1;
uint32 seed = 2;
}
/**
* Represents a continuous update from an image generation stream.
*/
message GenerateImagesStreamUpdate {
uint32 current_batch = 1;
oneof update {
GenerateImagesBatchProgressUpdate batch_progress = 2;
GenerateImagesBatchCompletedUpdate batch_completed = 3;
}
float overall_percentage_complete = 4;
}
/**
* The image generation service, for generating images from loaded models.
*/
@ -278,4 +301,6 @@ service ImageGenerationService {
* Generates images using a loaded model.
*/
rpc GenerateImages(GenerateImagesRequest) returns (GenerateImagesResponse);
rpc GenerateImagesStreaming(GenerateImagesRequest) returns (stream GenerateImagesStreamUpdate);
}