fix train path

pull/852/head
Jingyi 2024-07-18 11:53:32 +08:00
parent 55d9a258fc
commit cc4d5eed9c
2 changed files with 1 additions and 1 deletions

Binary file not shown.

View File

@ -101,7 +101,7 @@ def _trigger_sagemaker_training_job(
""" """
site_packages_s3_path = (f"s3://aws-gcr-solutions-{region}/" site_packages_s3_path = (f"s3://aws-gcr-solutions-{region}/"
f"stable-diffusion-aws-extension-github-mainline/{esd_version}/train.tar") f"stable-diffusion-aws-extension-github-mainline/{esd_version}/site-packages.tar")
data = { data = {
"id": train_job.id, "id": train_job.id,