daniel543x/artcraft
1
0 Fork 0
mirror of https://github.com/storytold/artcraft.git synced 2026-11-06 02:00:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

parameter for predefined SD weights

Browse Source
This commit is contained in:
Brandon Thomas committed 2024-02-02 23:17:12 -05:00
1 parent 0540d2caba
commit c8fa722b3d
2 files changed
+8 -1
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

crates/service/job/inference_job/src/job/job_types/image_generation/sd/process_job.rs
+4 -1
View File
@@ -416,7 +416,10 @@ pub async fn process_job_lora(
// download model + vae for lora
// use lora with whatever checkpoint make it fixed id if it works submit it
let weight_token_sd_model_token = ModelWeightToken(String::from("weight_dmmthavhawqc2hj7yqyemcbf8")); // any generic model.
let weight_token_sd_model_token = ModelWeightToken::new_from_str(&sd_deps.predefined_sd_weight_token);
info!("Using predefined SD weight token: {:?}", &weight_token_sd_model_token);
let sd_weight_record = get_weight_by_token(
&weight_token_sd_model_token,
false,
crates/service/job/inference_job/src/job/job_types/image_generation/sd/stable_diffusion_dependencies.rs
+4
View File
@@ -8,6 +8,8 @@ pub struct StableDiffusionDependencies {
pub downloaders: StableDiffusionDownloaders,
pub inference_command: StableDiffusionInferenceCommand,
pub vae_bucket_path: String,
pub predefined_sd_weight_token: String,
pub predefined_lora_weight_token: String,
}
impl StableDiffusionDependencies {
@@ -16,6 +18,8 @@ impl StableDiffusionDependencies {
downloaders: StableDiffusionDownloaders::build_all_from_env(),
inference_command: StableDiffusionInferenceCommand::from_env()?,
vae_bucket_path: easyenv::get_env_string_required("SD_VAE_BUCKET_PATH")?,
predefined_sd_weight_token: easyenv::get_env_string_required("SD_PREDEFINED_SD_WEIGHT_TOKEN")?,
predefined_lora_weight_token: easyenv::get_env_string_required("SD_PREDEFINED_LORA_WEIGHT_TOKEN")?,
})
}
}