From 69e6a73ef4bd0bf2fcecfc4bf3916cb18f53d399 Mon Sep 17 00:00:00 2001 From: ArEnSc Date: Thu, 25 Jan 2024 13:46:30 -0500 Subject: [PATCH] Fixing issues and testing. --- .../endpoints/image_gen/enqueue_image_generation.rs | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/crates/service/web/storyteller_web/src/http_server/endpoints/image_gen/enqueue_image_generation.rs b/crates/service/web/storyteller_web/src/http_server/endpoints/image_gen/enqueue_image_generation.rs index 5ce05f87b7..eb6ada914e 100644 --- a/crates/service/web/storyteller_web/src/http_server/endpoints/image_gen/enqueue_image_generation.rs +++ b/crates/service/web/storyteller_web/src/http_server/endpoints/image_gen/enqueue_image_generation.rs @@ -114,6 +114,7 @@ pub struct EnqueueImageGenRequest { maybe_batch_count: Option, maybe_name: Option, maybe_description: Option, + maybe_version: Option } #[derive(Serialize, ToSchema)] @@ -369,10 +370,15 @@ pub async fn enqueue_image_generation_request( seed = s; } + let mut version:u32 = 0; let type_of_inference = mode.to_string().clone(); let description = request.maybe_description.clone(); let name = request.maybe_name.clone(); + if let Some(s) = request.maybe_version { + version = s; + } + let inference_args = StableDiffusionArgs { maybe_sd_model_token: Some(sd_weight_token), maybe_lora_model_token: Some(lora_token), @@ -390,6 +396,7 @@ pub async fn enqueue_image_generation_request( maybe_sampler: Some(sampler), maybe_description: description, maybe_name: name, + maybe_version: version, }; // create the inference args here