Add template test files.

This commit is contained in:
Michael Chung committed 2023-12-11 21:53:25 -05:00
1 parent a33ed7c897
commit 0b4e3a6640
11 files changed
+16 -1

No files matched your search

@@ -112,6 +112,17 @@ pub async fn dispatch_job_to_handler<'a, 'b: 'a>(args: DispatchJobToHandlerArgs<
).await?;
entity_token = results.entity_token.clone();
entity_type = results.entity_type.clone();
},
GenericDownloadType::StableDiffusion15 => {
let results = process_vits_model(
args.job_runner_state,
args.job,
args.temp_dir,
args.download_filename,
args.redis_logger,
).await?;
entity_token = results.entity_token.clone();
entity_type = results.entity_type.clone();
}
}
Whitespace-only changes.
Whitespace-only changes.
@@ -0,0 +1,5 @@
pub mod model_downloaders;
pub mod pretrained_sd15_model;
pub mod process_job;
pub mod sd15_dependencies;
pub mod sd15_inference_command;
@@ -149,7 +149,6 @@ pub async fn upload_weights_handler(
let job_token;
let download_job_type;
let generic_download_type = match supported_model_weights_type {
WeightsType::HifiGan => GenericDownloadType::HifiGan,
WeightsType::RvcV2 => GenericDownloadType::RvcV2,