diff --git a/crates/service/job/inference_job/config/inference-job.development.env b/crates/service/job/inference_job/config/inference-job.development.env index 89609b0429..9946620e5d 100644 --- a/crates/service/job/inference_job/config/inference-job.development.env +++ b/crates/service/job/inference_job/config/inference-job.development.env @@ -124,7 +124,7 @@ RERENDER_INFERENCE_EXECUTABLE_OR_COMMAND="venv/bin/python rerender.py" STABLE_DIFFUSION_INFERENCE_ROOT_DIRECTORY="/home/tensor/code/storyteller/storyteller-ml/image_generation/Storyteller-Studio-Filter/" STABLE_DIFFUSION_INFERENCE_COMMAND=".venv/bin/python main.py" SD_VAE_BUCKET_PATH="/stable_diffusion_deps/vaeFtEma560000Ema_original.pt" -SD_PREDEFINED_SD_WEIGHT_BUCKET_PATH="TODO" +SD_PREDEFINED_SD_WEIGHT_BUCKET_PATH="/stable_diffusion_deps/fiamixRebootPastel_v14LightColor.safetensors" # TODO(bt,2024-02-10): Remove these SD_PREDEFINED_SD_WEIGHT_TOKEN="INVALID_VALUE_REMOVE" diff --git a/crates/service/job/inference_job/src/job/job_types/image_generation/sd/process_job_lora_upload.rs b/crates/service/job/inference_job/src/job/job_types/image_generation/sd/process_job_lora_upload.rs index 4f77f4adf8..efff3bab07 100644 --- a/crates/service/job/inference_job/src/job/job_types/image_generation/sd/process_job_lora_upload.rs +++ b/crates/service/job/inference_job/src/job/job_types/image_generation/sd/process_job_lora_upload.rs @@ -164,6 +164,10 @@ pub async fn process_job_lora_upload( // let remote_cloud_file_client = RemoteCloudFileClient::get_remote_cloud_file_client().await?; // remote_cloud_file_client.download_file(sd_weight_details, path_to_string(sd_checkpoint_path.clone())).await?; + info!("Downloading predefined SD weight from: {:?} to {:?}", + &sd_deps.predefined_sd_weight_bucket_path, + &sd_checkpoint_path); + args.job_dependencies .buckets .public_bucket_client @@ -174,6 +178,10 @@ pub async fn process_job_lora_upload( ProcessSingleJobError::from_anyhow_error(anyhow!("could not download predefined SD weight: {:?}", err)) })?; + info!("Downloading predefined VAE weight from: {:?} to {:?}", + &sd_deps.vae_bucket_path, + &vae_path); + args.job_dependencies .buckets .public_bucket_client