Fixing issues and testing.

This commit is contained in:
ArEnSc committed 2024-01-25 13:46:30 -05:00
1 parent 51ce0c29cb
commit 69e6a73ef4
1 file changed
+7
@@ -114,6 +114,7 @@ pub struct EnqueueImageGenRequest {
maybe_batch_count: Option<u32>,
maybe_name: Option<String>,
maybe_description: Option<String>,
maybe_version: Option<u32>
}
#[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