log downloads

This commit is contained in:
Brandon Thomas committed 2024-02-10 17:10:33 -05:00
1 parent 5fa8bb93ce
commit f16ca22110
2 files changed
+9 -1

No files matched your search

@@ -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"
@@ -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